Python 文件操作:获取文件列表
当我们分析某个工作或机器学习项目进行数据处理时,需要获取特定目录中的文件列表。
通常,文件名具有匹配的模式。假设我们要查找目录中的所有.txt文件,可使用Path对象的方法glob()来实现。glob()方法创建了一个生成器,允许我们进行迭代。
>>> txt_files = list(Path('.').glob("*.txt"))
... print("Txt files:", txt_files)
...
Txt files: [PosixPath('hello_world.txt'), PosixPath('hello.txt')]
另外,直接使用glob模块也很方便,如下所示,通过创建可以使用的文件名列表,它具有相似的功能。在大多数情况下,例如文件读取和写入,两者都可以使用。
>>> from glob import glob
... files = list(glob('h*'))
... print("以h开头的文件:", files)
...
Files starting with h: ['hello_world.txt', 'hello.txt']