协慌网

登录 贡献 社区

在 Python 中查找扩展名为. txt 的目录中的所有文件

如何在 python 中找到扩展名为.txt的目录中的所有文件?

答案

你可以使用glob

import glob, os
os.chdir("/mydir")
for file in glob.glob("*.txt"):
    print(file)

或者只是os.listdir

import os
for file in os.listdir("/mydir"):
    if file.endswith(".txt"):
        print(os.path.join("/mydir", file))

或者如果要遍历目录,请使用os.walk

import os
for root, dirs, files in os.walk("/mydir"):
    for file in files:
        if file.endswith(".txt"):
             print(os.path.join(root, file))

使用glob

>>> import glob
>>> glob.glob('./*.txt')
['./outline.txt', './pip-log.txt', './test.txt', './testingvim.txt']

这样的事情应该可以胜任

for root, dirs, files in os.walk(directory):
    for file in files:
        if file.endswith('.txt'):
            print file