Поиск папок в каталоге (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