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

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

1 基本流程

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

图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)