📅  最后修改于: 2023-12-03 14:51:21.384000             🧑  作者: Mango
在Python中,我们可以使用os模块和os.walk()函数来递归遍历目录及其子目录,查找具有特定扩展名的文件。
以下是一个示例代码,它可以查找指定目录下所有的.py文件:
import os
def find_files(directory, extension):
# 遍历目录及其子目录
for root, dirs, files in os.walk(directory):
for file in files:
# 判断文件是否具有指定的扩展名
if file.endswith(extension):
print(os.path.join(root, file))
在这个示例中,我们定义了一个find_files()函数,它接受两个参数:要查找的目录和扩展名。
我们使用os.walk()函数来遍历目录及其子目录。对于每个找到的文件,我们使用endswith()函数来检查它们是否具有指定的扩展名。如果是,我们使用os.path.join()函数构建文件的全路径并打印出来。
你可以调用这个函数来查找指定目录下所有的.py文件:
find_files('/path/to/directory', '.py')
你也可以修改这个函数来查找其他扩展名的文件:
find_files('/path/to/directory', '.txt')
find_files('/path/to/directory', '.csv')
通过上面的示例,你学习了如何在Python中递归遍历目录及其子目录,查找具有特定扩展名的文件。你可以根据自己的需求修改代码以查找其他文件类型。