如何使用Python查找目录中具有扩展名.txt的所有文件?

在日常的编程工作中，我们经常需要查找特定类型的文件。对于扩展名为txt的文件，Python提供了简单易用的方法来查找并读取这些文件。下面我们将介绍如何使用Python查找目录中以txt为扩展名的所有文件。

方法一：使用os模块和列表推导式

我们可以使用os模块中的walk方法来遍历整个目录树，然后使用列表推导式来快速筛选出所有以”.txt”结尾的文件。

import os

def find_txt_files(dir_path):
    txt_files = [os.path.join(root, file) for root, dirs, files in os.walk(dir_path) for file in files if file.endswith('.txt')]
    return txt_files

txt_files = find_txt_files('/path/to/directory')
print(txt_files)

我们首先定义了一个find_txt_files函数，它接受一个参数dir_path，代表要查找的目录路径。函数中使用了os模块的walk方法来遍历整个目录树，并使用列表推导式来筛选出所有以”.txt”结尾的文件。

调用函数后，将会返回一个包含所有找到的txt文件路径的列表。可以根据需要对这个列表进行处理，比如逐个读取这些文件的内容。

方法二：使用glob模块

除了使用os模块，我们也可以使用Python的内置模块glob来查找目录中的文件。glob模块提供了一个名为glob的函数，可以根据用户给定的模式来筛选文件名。比如我们可以使用glob('*.txt')来搜索当前目录中以”.txt”结尾的文件。

import glob

def find_txt_files(dir_path):
    txt_files = glob.glob(os.path.join(dir_path, '*.txt'))
    return txt_files

txt_files = find_txt_files('/path/to/directory')
print(txt_files)

上述代码中，我们定义了一个find_txt_files函数，它接受一个参数dir_path，代表要查找的目录路径。函数中使用glob函数和一个模式*.txt来筛选所有以”.txt”结尾的文件。

调用函数后，将会返回一个包含所有找到的txt文件路径的列表。

方法三：使用pathlib模块

Python 3.4及以上版本还引入了一个新的库”Pathlib”，它的设计理念是将文件路径封装成对象，并提供一些简单易用的方法来操作这些对象。使用Pathlib库，我们可以很方便地查找目录中的文件。具体实现方法如下：

from pathlib import Path

def find_txt_files(dir_path):
    p = Path(dir_path)
    txt_files = list(p.glob('**/*.txt'))
    return txt_files

txt_files = find_txt_files('/path/to/directory')
print(txt_files)

Pathlib库中提供了一个名为Path的类来表示文件路径。我们可以使用glob方法来遍历整个目录树，并使用一个类似于正则表达式的模式来筛选出所有以”.txt”结尾的文件。

调用函数后，将会返回一个包含所有找到的txt文件路径的列表。