Идиома 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)
Конструкция * распаковывается в кортеж, чтобы передать аргументы, поэтому нет возможности использовать его.