极客笔记Python 如何查找具有.txt扩展名的目录中的所有文件
使用Python中的工具可以轻松完成在目录中查找特定文件的任务；在某些情况下，您可能需要使用Python查找目录中具有.txt扩展名的所有文件。让我们深入了解此任务所涉及的过程，并为您提供多种以易于理解的代码示例和解释来实现在目录中查找具有.txt扩展名的所有文件的方法。
使用os.listdir()函数
在这个代码示例中，我们首先导入os模块，它是在Python中处理目录和文件的关键模块。
示例
find_txt_files()函数以directory_path作为其参数；directory_path表示要搜索的目录的路径。
我们使用os.listdir(directory_path)来获取指定目录中所有项的列表，即文件和目录。
通过对每个项进行迭代，并通过使用os.path.isfile()来检查是否为文件，我们确保仅考虑文件而不是目录。
在第二个条件中，我们使用item.endswith(‘.txt’)来获取只有.txt扩展名的文件。
该函数将目录中找到的文本文件列表作为输出。
import os
def find_txt_files(directory_path):
try:
# Use os.listdir() to obtain a list of all items in the directory
all_items = os.listdir(directory_path)
# Filter out only the files with '.txt' extension
txt_files = [item for item in all_items if
os.path.isfile(os.path.join(directory_path, item)) and item.endswith('.
txt')]
return txt_files
except FileNotFoundError:
print(f"Error: The directory '{directory_path}' does not exist.")
return []
# Replace 'directory_path' with the path of the directory you want to
search
directory_path = '/path/to/your/directory'
txt_files_list = find_txt_files(directory_path)
if txt_files_list:
print("Text files in the directory:")
for file_name in txt_files_list:
print(file_name)
else:
print("No .txt files found in the directory.")
输出
对于某个目录，获得了以下输出
Text files in the directory:
fubar.txt
使用os.listdir()
示例
在本例中，首先我们导入os模块，该模块使我们能够与操作系统、目录和文件进行交互。
find_txt_files()函数接受directory_path作为其参数。directory_path代表要搜索.txt文件的目录的路径。
使用os.listdir(directory_path)函数获取指定目录中所有项（即文件和目录）的列表。
通过迭代每个项并使用os.path.isfile()检查是否为文件，我们确保只考虑文件并忽略目录。
在第二个实例中，我们使用item.endswith(‘.txt’)来仅寻找具有.txt扩展名的文件。
该函数返回一个找到的目录中的.txt文件的列表。
import os
def find_txt_files(directory_path):
try:
# Get a list of all items (files and directories) in the specified
directory
all_items = os.listdir(directory_path)
# Filter out only the files with the '.txt' extension
txt_files = [item for item in all_items if os.path.isfile(os.path.
join(directory_path, item)) and item.endswith('.txt')]
return txt_files
except FileNotFoundError:
print(f"Error: The directory '{directory_path}' does not exist.")
return []
# Replace 'directory_path' with the path of the directory you want to
search
directory_path = '/path/to/your/directory'
txt_files_list = find_txt_files(directory_path)
if txt_files_list:
print("Text files in the directory:")
for file_name in txt_files_list:
print(file_name)
else:
print("No .txt files found in the directory.")
输出
对于某个目录，获得了以下输出
Text files in the directory:
fubar.txt
使用os.scandir()进行效率优化
示例
这里，os.listdir()被替换为os.scandir()，以提供一种更高效的方法来列出目录中的文件。
当将os.scandir(directory_path)的输出用作条目列表时，会创建一个上下文管理器，发现它能够高效地迭代遍历目录条目，并且不需要显式关闭目录。
通过使用entry.is_file()，检查每个条目是否为文件，如果是文件，则继续检查其是否以 .txt 结尾。
发现该函数返回一个在目录中找到的 .txt 文件列表。
import os
def find_txt_files(directory_path):
try:
# Use os.scandir() for a more efficient listing
with os.scandir(directory_path) as entries:
txt_files = [entry.name for entry in entries if entry.is_file()
and entry.name.endswith('.txt')]
return txt_files
except FileNotFoundError:
print(f"Error: The directory '{directory_path}' does not exist.")
return []
# Replace 'directory_path' with the path of the directory you want to
search
directory_path = '/path/to/your/directory'
txt_files_list = find_txt_files(directory_path)
if txt_files_list:
print("Text files in the directory:")
for file_name in txt_files_list:
print(file_name)
else:
print("No .txt files found in the directory.")
输出
针对某个目录，获得了以下输出
Text files in the directory:
fubar.txt
递归搜索 – os.walk()
示例
在这个特定的示例中，我们使用os.walk()来对包括子目录在内的 .txt 文件进行递归搜索。
然后os.walk(directory_path)函数返回一个生成器，生成包含根目录、子目录和该目录中文件的元组。
每个元组都会被迭代，并且对于files列表中的每个文件，我们使用file.endswith(‘.txt’)来判断是否以 .txt 扩展名结尾。
如果确实以该扩展名结尾，我们使用os.path.join(root, file)构建完整的文件路径，并将该文件添加到txt_files列表中。
最终，该函数返回一个包含所找到的目录及其子目录中的所有 .txt 文件的综合列表。
import os
def find_txt_files(directory_path):
try:
# Use os.walk() to get a recursive listing of all files
txt_files = []
for root, dirs, files in os.walk(directory_path):
for file in files:
if file.endswith('.txt'):
txt_files.append(os.path.join(root, file))
return txt_files
except FileNotFoundError:
print(f"Error: The directory '{directory_path}' does not exist.")
return []
# Replace 'directory_path' with the path of the directory you want to
search
directory_path = '/path/to/your/directory'
txt_files_list = find_txt_files(directory_path)
if txt_files_list:
print("Text files in the directory:")
for file_name in txt_files_list:
print(file_name)
else:
print("No .txt files found in the directory.")
输出
对于某个目录，获得了以下的输出结果。
Text files in the directory:
/content/foo/fubar.txt
使用pathlib.Path()进行现代列举
示例
在这个最后的例子中，我们采用了最新和现代的方法，通过使用pathlib.Path()来执行相同的列举任务。
我们首先从pathlib模块导入Path；该模块提供了一个面向对象的接口，用于操作目录和文件。
通过使用Path(directory_path)创建一个指向指定目录的Path对象。
使用path.iterdir()创建一个包括文件和目录的所有条目的迭代器，以获得目录内的相同内容。
使用file.is_file()函数检查每个条目是否为文件，如果确实是文件，则通过使用file.suffix检查是否具有.txt后缀。
如果所有这些条件都满足，则将文件包含在txt_files列表中。
然后发现该函数返回一个在目录中找到的.txt文件列表。
from pathlib import Path
def find_txt_files(directory_path):
try:
# Use pathlib.Path() for modern file listing
path = Path(directory_path)
txt_files = [file for file in path.iterdir()
if file.is_file() and file.suffix == '.txt']
return txt_files
except FileNotFoundError:
print(f"Error: The directory '{directory_path}'
does not exist.")
return []
# Replace 'directory_path' with the path of the
directory you want to search
directory_path = '/path/to/your/directory'
txt_files_list = find_txt_files(directory_path)
if txt_files_list:
print("Text files in the directory:")
for file_name in txt_files_list:
print(file_name)
else:
print("No .txt files found in the directory.")
输出
对于某个目录，得到了以下输出
Text files in the directory:
/content/foo/fubar.txt
这里有四种多样且高效的方法可以使用Python在目录中找到所有扩展名为.txt的文件。你总是可以选择任意一种或多种方式，比如经典的os.listdir()、高效的os.scandir()、递归的os.walk()，或者现代的pathlib.Path()，以满足你的特定需求。通过学习这些代码示例和解释，你现在拥有了一个多功能的工具包，可以用来自动搜索文件和优雅地组织你的Python项目。
通过练习这些简洁而优雅的代码片段，你可以轻松地在任何目录中找到所有扩展名为.txt的文件。Python的多功能性和易用性使其成为处理与文件相关的任务的绝佳选择，无论是管理数据，组织文件，还是处理文本文件进行分析。