📅  最后修改于: 2023-12-03 15:11:53.064000             🧑  作者: Mango
在Python程序开发中,有时候我们需要获取某个目录下的所有文件夹名称。本篇文章介绍了如何用Python获取目录中的文件夹列表。
Python标准库中的os
模块提供了获取目录中所有文件和文件夹列表的功能。下面是一个示例代码:
import os
# 获取当前目录下所有文件夹名称
folders = [name for name in os.listdir('.') if os.path.isdir(name)]
print(folders)
上述代码使用了os.listdir()
函数获取当前目录下的所有文件和文件夹名称,接着使用了os.path.isdir()
函数筛选出其中的文件夹名称。最终输出所有获取到的文件夹名称。
注意,os.listdir()
返回的是当前目录下所有文件和文件夹的名称,包括.
和..
等特殊文件夹,故需要使用os.path.isdir()
进行筛选。
除了os
模块外,Python还提供了pathlib
模块,其提供了更加面向对象的方式来操作文件和文件夹路径。下面是使用pathlib
模块实现的示例代码:
import pathlib
# 获取当前目录下所有文件夹名称
folders = [p.name for p in pathlib.Path('.').iterdir() if p.is_dir()]
print(folders)
上述代码使用了pathlib.Path()
函数获取当前目录下的所有文件和文件夹路径,接着使用了iterdir()
函数遍历并筛选其中的文件夹路径,最终输出所有获取到的文件夹名称。
需要注意的是,pathlib.Path().name
方法返回的是路径中的最后一部分(即文件或文件夹名称),所以可以直接获取文件夹名称。
通过上面的介绍,我们可以看出,无论是使用os
模块还是pathlib
模块,都可以方便地获取目录中的文件夹列表。建议读者可以尝试在项目中使用这些代码片段。