python中的os.listdir()与os.path.isdir()

python中的os模块提供了丰富的方法来处理文件和目录。

1.os.listdir(path)

是用来返回path指定的文件夹包含的文件或文件夹的名字的列表。
下面给出一个例子：

图中是我文件夹下的文件，运行下面的代码

import os

path = './data_set/flower_data/flower_photos'
for i in os.listdir(path):
    print(i)

可以看到结果如下：

2.os.path.isdir()

是用来判断路径是否为目录
还是上面的文件夹，运行如下代码：

import os

path = './data_set/flower_data/flower_photos'
for i in os.listdir(path):
    if os.path.isdir(os.path.join(path, i)):
        print(i)

结果为：

可以对比看到，此时的输出没有txt文件，所以说isdir()是用来判断此路径是否为一个目录，简单理解为是否是一个可展开的文件夹的。
注意这里要使用绝对路径，因此又使用了os.path.join()函数，这个函数的作用就是将两个路径拼接在一起。