Идиома Python для цепочки (сглаживания) бесконечного итерируемого конечных итераций?

Предположим, что у нас есть итератор (бесконечный), который возвращает списки (или конечные итераторы), например, один, возвращенный

infinite = itertools.cycle([[1,2,3]])

Что такое хорошая идиома Python для получения итератора (очевидно, бесконечного), который вернет каждый из элементов из первого итератора, затем каждый из второго и т.д. В приведенном выше примере он вернет 1,2,3,1,2,3,.... Итератор бесконечен, поэтому itertools.chain(*infinite) не будет работать.

Связанные

Ответы

Ответ 1

Начиная с Python 2.6, вы можете использовать itertools.chain.from_iterable:

itertools.chain.from_iterable(iterables)

Вы также можете сделать это с помощью вложенного генератора:

def flatten(iterables):
    return (elem for iterable in iterables for elem in iterable)

Ответ 2

Используйте генератор:

(item for it in infinite for item in it)

Конструкция * распаковывается в кортеж, чтобы передать аргументы, поэтому нет возможности использовать его.