问题

如何列举一个目录下的所有文件

解决方案

使用os.listdir()

>>> import os
>>> os.listdir('.')
['.android', '.BigNox', '.CLion2017.3', '.PyCharm2017.3', '.swt', '3D Objects', 'AppData', 'Application Data']

如果只需要文本文件的话,就结合os.path.isfile进行过滤

>>> [f for f in os.listdir('.') if os.path.isfile(f)]
['inittk.ini', 'inst.ini', 'NTUSER.DAT', 'ntuser.dat.LOG1', 'ntuser.dat.LOG2']

如果对文件有匹配需求的话,比如只需要某个目录的txt文件,那么可以使用glob模块,例如:

import glob
print(glob.glob("/home/adam/*.txt"))

讨论

os.listdir()会返回一个列表,包含目录下的所有文件,但不会包含...,即使他们是存在的。

glob提供了Unix风格的路径名匹配,支持*?[]匹配,并且glob函数还包含一个可选的参数recursive决定是否递归搜索,下面是一个glob函数的例子

>>> import glob
>>> glob.glob('./[0-9].*')
['./1.gif', './2.txt']
>>> glob.glob('*.gif')
['1.gif', 'card.gif']
>>> glob.glob('?.gif')
['1.gif']
>>> glob.glob('**/*.txt', recursive=True)
['2.txt', 'sub/3.txt']
>>> glob.glob('./**/', recursive=True)
['./', './sub/']

来源

Stack Overflow

关注

欢迎关注我的微信公众号:python每日一练

本文固定链接: http://www.js-code.com/python/python_49622.html