实现遍历文件夹并查找后缀为".py"的文件,可以采用递归算法,即先找到当前目录下所有文件和文件夹,如果是文件则判断后缀是否为".py",如果是文件夹则继续递归查找子目录,直到找到所有符合条件的文件为止。
以下是具体步骤:
步骤一:导入必要的模块
Python自带的os模块提供了一些用于文件和目录处理的函数,需要先导入该模块。
import os
步骤二:定义函数
我们可以定义一个名为find_py_files
的函数完成文件夹遍历的过程。该函数接受一个参数,表示要查找的目录路径。在函数内部,我们可以使用os.listdir()
函数获取path
路径下的所有文件和文件夹。
def find_py_files(path):
# 获取目录下所有的文件和目录
filenames = os.listdir(path)
# 遍历每一个文件和目录
for filename in filenames:
# 拼接成文件或目录的完整路径
full_path = os.path.join(path, filename)
# 判断是否为文件
if os.path.isfile(full_path):
# 判断是否为.py文件
if full_path.endswith('.py'):
# 打印文件路径
print(full_path)
# 判断是否为目录
elif os.path.isdir(full_path):
# 递归查找子目录
find_py_files(full_path)
示例一:查找当前目录下的.py文件
可以通过传入'.'作为参数调用该函数来查找当前目录下的".py"文件。
find_py_files('.')
执行以上代码,控制台会输出所有找到的.py文件的路径:
./test.py
./utils/myutil.py
示例二:查找指定目录下的.py文件
也可以直接将指定的路径作为参数传入该函数来查找该路径下的".py"文件。例如,要查找"/data/code/"目录下的.py文件,可以这样调用:
find_py_files('/data/code/')
这样,函数就会在/data/code/目录下对所有文件和子目录进行递归遍历,并输出其中所有的.py文件路径。
注意:在Linux和MacOS系统中,目录分隔符为"/",而在Windows系统中,目录分隔符为"\",因此,在使用Windows系统时,可能需要手动将目录分隔符改为"\"。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:python遍历文件夹找出文件夹后缀为py的文件方法 - Python技术站