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

时间:2017-12-15 21:49:59

标签: python directory filepath glob python-os

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

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

1 个答案:

答案 0 :(得分:4)

您可以使用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))