Python每日一练0025

问题

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

解决方案

使用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每日一练