Как клонировать объект генератора Python?

Рассмотрим этот сценарий:

  #! / usr / bin / env python
 # - * - кодирование: utf-8 - * -
 import os

 walk = os.walk ('/ home')

 для root, dirs, файлов в пути:
     для имени пути в файлах dirs +:
         print os.path.join (root, pathname)

 для root, dirs, файлов в пути:
     для имени пути в файлах dirs +:
         print os.path.join (root, pathname)

Я знаю, что этот пример является излишним, но вы должны учитывать, что нам нужно использовать одни и те же данные walk более одного раза. У меня есть эталонный сценарий, и использование одинаковых данных walk является обязательным для получения полезных результатов.

Я пробовал walk2 = walk к клону и использовать на второй итерации, но это не сработало. Вопрос … Как я могу его скопировать? Возможно ли это?

Заранее спасибо.

Вы можете использовать itertools.tee() :

 walk, walk2 = itertools.tee(walk)

Обратите внимание, что это может «потребовать значительного дополнительного хранения», как указано в документации.

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

walk = list(os.walk('/home'))

Определить функцию

  def walk_home(): for r in os.walk('/home'): yield r

Или даже это

 def walk_home(): return os.walk('/home')

Оба используются следующим образом:

 for root, dirs, files in walk_home(): for pathname in dirs+files: print os.path.join(root, pathname)

Этот ответ направлен на расширение / уточнение того, что высказали другие ответы. Решение обязательно будет зависеть от того , чего именно вы стремитесь достичь.

Если вы хотите os.walk один и тот же результат os.walk несколько раз, вам нужно будет инициализировать список из os.walk os.walk iterable (т. os.walk Walk walk = list(os.walk(path)) ).

Если вы должны гарантировать, что данные остаются теми же, это, вероятно, ваш единственный вариант. Однако существует несколько сценариев, в которых это невозможно или желательно.

Невозможно list() итерабельность, если выход имеет достаточный размер (т. Е. Попытка list() вся файловая система может заморозить ваш компьютер).
Нежелательно list() итерируемый, если вы хотите получить «свежие» данные перед каждым использованием.

Если list() не подходит, вам нужно будет запустить генератор по требованию. Обратите внимание, что генераторы гасятся после каждого использования, поэтому это создает небольшую проблему. Чтобы «повторно запустить» ваш генератор несколько раз, вы можете использовать следующий шаблон:

 #!/usr/bin/env python # -*- coding: utf-8 -*- import os class WalkMaker: def __init__(self, path): self.path = path def __iter__(self): for root, dirs, files in os.walk(self.path): for pathname in dirs + files: yield os.path.join(root, pathname) walk = WalkMaker('/home') for path in walk: pass # do something... for path in walk: pass

Вышеупомянутый шаблон дизайна позволит вам сохранить код DRY.

Как клонировать объект генератора Python?

Являются ли генераторы Threadsafe?

Инверсия словаря на месте в Python

Разница между генераторами Python и итераторами

Как работают Python Set Comprehensions?

Как использовать выражения генератора python для создания oneliner для запуска функции несколько раз и получения вывода списка

Есть ли способ сократить это выражение генератора Python?

Как создать генератор из двух списков в python?

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

безопасность потока генератора питона с использованием Keras

Разность скоростей между итерацией по генераторам и спискам