Как «открыть» список файлов и получить их ручки в python?

Question

Возможно ли до with open() все файлы, содержащиеся в списке, и создавать файлы для записи?

Например, если моя функция принимает список имен файлов для данных расщепления в задаче машинного обучения,

fname_list = ['train_dataset.txt', 'validate_dataset.txt', 'test_dataset.txt']

, то это было бы удобно, чтобы быть в состоянии сделать:

with open('source_dataset.txt) as src_file, open(name_list, 'w') as <DONT_KNOW_WHAT_TO_DO_HERE>: 

И выполните некоторые расщепления данных внутри блока.

Редактировать: Так что мой вопрос в основном «Можно ли получить несколько файлов для списка файлов, открытых с помощью« с открытым() »?

Answer 1

В Python 3.3 и выше, contextlib.ExitStack можно использовать, чтобы правильно и красиво сделать это:

from contextlib import ExitStack 

with open('source_dataset.txt') as src_file, ExitStack() as stack: 
    files = [stack.enter_context(open(fname, 'w')) for fname in fname_list] 
    ... do stuff with src_file and the values in files ... 
... src_file and all elements in stack cleaned up on block exit ... 

Answer 2

Вы можете определить класс openfiles поддержать with заявление:

class openfiles: 
    def __init__(self, filelist, mode='r'): 
     self.fhandles = [open(f, mode) for f in filelist] 

    def __enter__(self): 
     return self.fhandles 

    def __exit__(self, type, value, traceback): 
     map(file.close, self.fhandles) 

Тогда вам может:

with openfiles(['file1', 'file2']) as files: 
    for f in files: 
     print(f.read())