如何在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))
类似这样的东西应该可以完成工作
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.txt'):
print file