Python - для возврата генератора последовательностей

Question

Мне нужно сгенерировать функцию, которая принимает любое количество последовательностей и возвращает список кортежей. Я пытался писать код для генерации кортежей один за другим через генераторы из списка, используя следующий код:

>>> gen1 = [(x,y) for x in range(3) for y in range(4)] 
>>> gen1 
[(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (2, 0), (2, 1), (2, 2), (2, 3)] 
>>> iter1 = iter(gen1) 
>>> iter1.next() 
>>> def gen3(): 
...  yield iter1.next() 
... 
>>> next(gen3()) 

Который дает кортежи. Но мне нужно применить тот же код в следующей функции, которая содержит множество последовательностей параметров:

def generator_zip(seq1, seq2, *more_seqs): 

Как я могу использовать приведенный выше код в этой функции ??

Answer 1

Я думаю itertools.product является то, что вы ищете

import itertools 

print list(itertools.product(range(3), range(4))) 

>> [(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), 
>> (2, 0), (2, 1), (2, 2), (2, 3)] 

print list(itertools.product(range(2), range(3), range(4))) 

>> [(0, 0, 0), (0, 0, 1), (0, 0, 2), (0, 0, 3), (0, 1, 0), (0, 1, 1), 
>> (0, 1, 2), (0, 1, 3), (0, 2, 0), (0, 2, 1), (0, 2, 2), (0, 2, 3), 
>> (1, 0, 0), (1, 0, 1), (1, 0, 2), (1, 0, 3), (1, 1, 0), (1, 1, 1), 
>> (1, 1, 2), (1, 1, 3), (1, 2, 0), (1, 2, 1), (1, 2, 2), (1, 2, 3)] 

Обратите внимание, что itertools функция всегда возвращает генераторы