如何获取子目录名称列表

Question

有一个目录包含文件夹以及不同格式的文件.

import os 
my_list = os.listdir('My_directory')

将返回文件和文件夹名称的完整内容.例如,我可以使用endswith('.txt')方法来选择文本文件名,但是如何获取只是文件夹名称的列表？

Answer 1

在组装列表时尝试检查目录.

import os
output = [dI for dI in os.listdir('foo') if os.path.isdir(os.path.join('foo',dI))]

Answer 2

只需os.path.isdir在 返回的结果上使用os.listdir，如下所示：

def listdirs(path):
    return [d for d in os.listdir(path) if os.path.isdir(os.path.join(path, d))]

Answer 3

您可以使用 os.walk

for i,j,y in os.walk('.'):
    print(i)