Когда я пишу два файла в python, второй всегда пуст

Question

Я работаю над простым скриптом python для записи некоторых данных в два текстовых файла. Мой код выглядит так:

with open('file1.txt', 'w') as file1: 
    for thing in stuff1: 
     file1.write(thing + '\n') 

with open('file2.txt', 'w') as file2: 
    for thing in stuff2: 
     file2.write(thing + '\n') 

Когда я запускаю программу, файл1 получается как ожидалось, но file2 пуст.

Однако, если я переключу порядок, в котором я пишу в файлы (это означает, что я пишу сначала в файл2, а затем в файл1), файл2 получается как ожидалось, а файл1 пуст.

Другими словами, файл, в который я пишу второй, всегда оказывается пустым.

Я попытался добавить file1.flush()/file2.flush() внутри циклов, но безрезультатно. Любая идея, почему это может произойти?

Answer 1

Я угадываю здесь, но ваша проблема в том, что stuff1 и stuff2 указывают на тот же генератор или итератор, и вы можете только перебирать один раз. После этого его содержимое исчерпывается.

Попробуйте вместо этого:

import itertools 

stuff1, stuff2 = itertools.tee(the_source_generator_or_iterator_or_whatever) 

with open('file1.txt', 'w') as file1: 
    for thing in stuff1: 
     file1.write(thing + '\n') 

with open('file2.txt', 'w') as file2: 
    for thing in stuff2: 
     file2.write(thing + '\n') 

Answer 2

Я думаю, что у вас была линия

stuff2 = stuff1 

перед вашими фрагментами кода и stuff1, stuff2 являются генераторами. Если это так, они будут алиасами для одного и того же генератора. Первый цикл записи исчерпывает (одиночный) генератор.

Answer 3

Я думаю, что stuff2 пуст. Вы можете распечатать переменную для отладки перед тем, как перейти во второй цикл for.