使用多个文件扩展名来查找文件

Posted 2021-04-17

我正在使用glob通过这行代码列出主目录中的所有python文件。我想找到所有.json文件以及py文件，但我找不到任何在一个行代码中扫描多种文件类型。

for file in glob.glob('/home/mohan/**/*.py', recursive=True):
    print(file)

答案

您可以使用os.walk，它也可以在子目录中查找。

import os

for root, dirs, files in os.walk("path/to/directory"):
    for file in files:
        if file.endswith((".py", ".json")): # The arg can be a tuple of suffixes to look for
            print(os.path.join(root, file))