使用Python从多个文件夹中获取文件

Question

我需要使用具有以下结构的数据集构建一个包含一些文件的数组：

1. FolderA：

   • FolderA1：
   • FolderA2：
     • File01.txt

2. FolderB中：

   • FolderB1：
   • FolderB2：
     • File01.txt

假设我需要文件夹A，B，D和F中的File01.txt。如何获取这些文件？

Answer 1

您可以使用glob模块：

import glob

base_dir = '.'
folders = ('A', 'B', 'D', 'F')
pattern = '{}/[{}]/*/File01.txt'.format(base_dir, ''.join(folders))
for filename in glob.glob(pattern):
    print(filename)

创建glob模式./[ABDF]/*/File01.txt。如果脚本在示例中包含顶级文件夹的目录中执行，则会输出：

./A/A2/File01.txt
./B/B2/File01.txt

您还可以使用相对路径和绝对路径;只需根据需要更改base_dir。然后你可以从任何地方执行脚本。