Как открыть одновременно два файла с одинаковым именем и разным расширением в python?

Question

У меня есть папка с несколькими пару файлов:

a.txt 
    a.json 

    b.txt 
    b.json 

и так далее:

Использование для цикла я хочу открыть несколько файлов (a.txt и a.json) одновременно.

Есть ли способ сделать это, используя оператор 'with' в python?

Answer 1

Вы можете сделать что-то вроде следующего, которое строит словарь с ключом по имени файла sans extension и с подсчетом количества файлов, соответствующих требуемым расширениям. Затем вы можете перебирать пары файлов открытия словарей:

import os 
from collections import defaultdict 

EXTENSIONS = {'.json', '.txt'} 

directory = '/path/to/your/files' 

grouped_files = defaultdict(int) 

for f in os.listdir(directory): 
    name, ext = os.path.splitext(os.path.join(directory, f)) 
    if ext in EXTENSIONS: 
     grouped_files[name] += 1 

for name in grouped_files: 
    if grouped_files[name] == len(EXTENSIONS): 
     with open('{}.txt'.format(name)) as txt_file, \ 
       open('{}.json'.format(name)) as json_file: 
      # process files 
      print(txt_file, json_file)