你可以使用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