Python中获取指定目录下所有文件名的方法_Python

Python中获取指定目录下所有文件名的方法

2023-10-15 12:07棉猴 Python

在《Python中文件名和路径的操作》中提到，os模块中的函数可以对文件进行操作。通过递归以及os模块中提供的函数，可以获取指定目录下所有的文件名。

1 基本流程

通过递归获取指定目录下所有文件名的基本流程，如图1所示。

Python中获取指定目录下所有文件名的方法

图1 基本流程

2 函数实现

2.1 定义函数

定义一个名为walk的函数，该函数有一个参数dirname，表示指定的当前目录，代码如下所示：

def walk(dirname):

2.2 获取并访问当前目录下的所有子文件夹名及文件名

在《Python中文件名和路径的操作》中提到，可以通过os.listdir()函数获取指定目录下的所有子文件夹名及文件名，因此，在walk()函数内容，有如下代码：

for name in os.listdir(dirname):

以上代码通过for循环访问了os.listdir()函数获取到的所有子文件夹名和文件名，并将其保存到变量name中。

2.3 获取文件的绝对路径

在for循环内部，通过os.path.join()函数，获取文件的绝对路径，代码如下所示：

path = os.path.join(dirname, name)

其中，os.path.join()函数作用是将文件夹名和文件名进行连接，参数dirname是walk()函数的参数，即是文件夹名；参数name是os.listdir()函数获取的子文件夹名或文件名。最后，获取到的绝对路径保存在变量path中。

2.4 判断是否是文件

在for循环内部，获取到了绝对路径path之后，接下来判断path是否是文件，代码如下所示：

if os.path.isfile(path):
   print(path)
else:
   walk(path)

以上代码中，如果path是文件，则打印出该文件名；如果path不是文件，则说明是子文件夹，递归调用walk()函数，显示path子文件夹中的所有文件名。

3 函数调用

要想获取当前Python源文件所在目录的所有文件名，可以通过如下代码实现：

cwd = os.getcwd()
walk(cwd)

其中，os.getcwd()函数获取到当前Python源文件所在目录，将该目录作为参数传递给walk()函数，获取到该目录下的所有文件名。

4 完整代码

该程序的完整代码如下所示：

import os

def walk(dirname):
    for name in os.listdir(dirname):
        path = os.path.join(dirname, name)

        if os.path.isfile(path):
            print(path)
        else:
            walk(path)

cwd = os.getcwd()
walk(cwd)

到此这篇关于Python中获取指定目录下所有文件名的方法的文章就介绍到这了,更多相关内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文地址：https://blog.csdn.net/hou09tian/article/details/131498580