使用Python從多個文件夾中獲取文件

Question

我需要使用具有如下結構的數據集中的一些文件構建一個數組：

1. 文件夾A：

   • 文件夾A1：
   • 文件夾A2：
     • File01.txt

2. 資料夾B：

   • 文件夾B1：
   • 文件夾B2：
     • 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 。 然后，您可以在任何地方執行腳本。