张赐荣 | Python 递归算法: 列举文件和文件夹

在开始之前,先来了解一下递归算法。
递归是一种算法设计技巧,可以递归地处理文件夹和文件,使用 Python 的 os 模块中的 os.listdir() 和 os.path.isdir() 函数可以方便地实现递归处理。

首先,我们需要使用 Python 的 os 模块来访问文件系统。os 模块提供了很多有用的函数,其中之一是 os.listdir(),它可以返回给定文件夹中的文件列表。
所以,我们可以使用以下代码来列出给定文件夹中的所有文件:
import os
def list_files(folder):
# 获取文件夹中的文件列表
files = os.listdir(folder)
# 遍历文件列表
for file in files:
# 输出文件名
print(file)
# 调用函数,列出当前文件夹中的所有文件
list_files('.')
这段代码中的 '.' 表示当前文件夹。
现在,我们来看看如何递归文件夹。
我们可以在上面的函数中添加一个循环,用来处理文件夹中的每一个文件。如果遇到一个文件夹,我们可以再次调用 list_files() 函数来处理这个文件夹。这样,我们就可以递归处理文件夹了。
下面是完整的代码:
import os
def list_files(folder):
# 获取文件夹中的文件列表
files = os.listdir(folder)
# 遍历文件列表
for file in files:
# 如果是文件夹,则递归处理
if os.path.isdir(file):
list_files(file)
# 否则,输出文件名
else:
print(file)
# 调用函数,列出当前文件夹及其子文件夹中的所有文件
list_files('.')
这段代码中的 os.path.isdir() 函数用来判断给定的路径是否是一个文件夹。如果是,就递归处理这个文件夹;否则,就输出文件名。
好了,现在你已经学会了如何使用 Python 递归文件夹和文件。希望这篇教程对你有帮助。
 

猜你喜欢

转载自blog.csdn.net/zcr_59186/article/details/128388751