在目录中查找文件夹（Python 3）

Question

我正在使用下面的代码来搜索目录中的所有文本文件。

import os

numOfFiles=0
files=[]

dir_path = os.path.dirname(os.path.realpath(__file__))

for file in os.listdir(dir_path):
    if file.endswith(".txt"):
        files.append(file)
        print(os.path.join(dir_path, file))
        numOfFiles=numOfFiles+1

如何使用我现有的代码找到变量dir_path 中的所有子目录？

Answer 1

您可以查看isdir():

import os

numOfFiles=0
files=[]

dir_path = os.path.dirname(os.path.realpath(__file__))

for file in os.listdir(dir_path):
    if file.endswith(".txt"):
        files.append(file)
        print(os.path.join(dir_path, file))
        numOfFiles=numOfFiles+1
    if os.path.isdir(file):
        # ...Do something