python - 在python中设置搜索程序的目录

Question

我正在尝试开发一个用于图像处理的 CNN。我的电脑上的一个单独的驱动器上存储了大约 130 个演出，并且我在导航一个简单的 python 搜索程序来搜索指定的目录时遇到了麻烦。我试图让它找到一堆随机 XML 文件，这些文件分散在该驱动器上的许多子目录/子目录/子目录中。如何为这个 python 程序指定它应该搜索的目录，并将其仅保留在程序的上下文中？

我尝试设置变量 Path = "B:\\MainFolder\SubFolder" 并使用 os.walk，但它通过第一个目录然后停止。

Answer 1

您可以尝试以下操作吗:

import os
import glob
base_dir = 'your/start/sirectory'
req_files = glob.glob(os.path.join(base_dir, '**/*.xml'), recursive=True)