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

Question

在Python中找到扩展名为.txt的目录中的所有文件如何找到具有扩展名的目录中的所有文件？.txt在蟒蛇？

Answer 1

你可以用glob:

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

或者简单的os.listdir:

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

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

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

Answer 2

诸如此类的事情应该能胜任这项工作。

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