如何使用Python查找目录中具有扩展名.txt的所有文件?
您可以使用os.listdir方法获取目录中的所有目录和文件。然后过滤列表以仅获取文件,并检查它们的扩展名。
阅读更多:Python 教程
例如:
>>> import os
>>> file_list = [f for f in os.listdir('.') if os.path.isfile(os.path.join('.', f)) and f.endswith('.txt')]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt']
endswith方法是字符串类的成员,用于检查字符串是否以某个后缀结尾。
您还可以使用glob模块达到相同的效果:
>>> import glob, os
>>> file_list = [f for f in glob.glob("*.txt")]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt']
极客教程