python-2.7 - 如何在python中同时打开两个具有相同名称和不同扩展名的文件？

Question

我有一个包含多个文件的文件夹:

  a.txt
  a.json

  b.txt
  b.json

等等:

我想使用 for 循环同时打开几个文件(a.txt 和 a.json)。

有没有办法在 python 中使用“with”语句来实现？

Answer 1

您可以执行类似以下操作的操作，它构造一个由文件名无扩展名键入的字典，并计算与所需扩展名匹配的文件数。然后你可以遍历打开文件对的字典:

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)