Python如何列出目录下所有文件

一、获取当前工作目录

程序在运行时，需要知道当前所在的工作目录，才能够列出该目录下的文件。Python提供了os模块来处理文件和目录的操作。os.getcwd()函数可以返回当前工作目录并存储到一个变量中。

import os

# 获取当前工作目录
cwd = os.getcwd()
print(cwd)  # 输出：/User/test/

上述代码中，我们使用os模块的getcwd()函数获取当前工作目录，并将其保存到变量cwd中。输出结果为当前工作目录的路径。

二、列出目录下的所有文件

在获取当前工作目录之后，我们需要使用os.listdir()函数列出该目录下的所有文件。该函数返回一个列表，其中包含文件夹和文件的名称。我们可以使用for循环遍历该列表，并使用os.path.join()函数将路径和文件名结合起来，以获取每个文件的完整路径。

import os

# 获取当前工作目录
cwd = os.getcwd()

# 列出目录下的所有文件
for filename in os.listdir(cwd):
    # 获取文件的完整路径
    file_path = os.path.join(cwd, filename)
    # 判断是否是文件
    if os.path.isfile(file_path):
        print(file_path) # 输出文件的完整路径
    elif os.path.isdir(file_path):
        print("文件夹：" + file_path) # 输出文件夹的路径

上述代码中，我们遍历了当前工作目录下的所有文件和文件夹，并使用os.path.join()函数将路径和文件名组合起来，以获得每个文件和文件夹的完整路径。我们使用os.path.isfile()函数判断该路径是否为文件，并输出完整路径。如果该路径为文件夹，则使用os.path.isdir()函数判断，并输出文件夹的路径。

三、递归地列出目录下的所有文件

在需要列出所有文件的子目录时，我们需要使用递归函数。一个递归函数是在函数内部调用自己，直到某个条件满足为止。在列出目录下的所有文件时，我们需要遍历当前目录下的所有文件和文件夹，并查看每个文件夹是否包含子目录，如果有，则继续遍历子目录中的文件。这里我们使用os.walk()函数，该函数递归地遍历目录结构，并返回包含当前目录中所有文件和文件夹名称的元组。

import os

def list_files(path):
    # 递归地列出目录下的所有文件
    for root, dirs, files in os.walk(path):
        for filename in files:
            # 获取文件的完整路径
            file_path = os.path.join(root, filename)
            print(file_path) # 输出文件的完整路径

# 获取当前工作目录
cwd = os.getcwd()

# 列出目录下的所有文件
list_files(cwd)

上述代码中，我们定义了一个名为list_files()的函数，该函数使用os.walk()函数递归遍历目录结构并列出指定路径下所有的文件路径。我们在最后调用该函数，并传入当前工作目录作为参数。

四、根据文件名后缀筛选文件

在实际应用中，我们可能需要只列出特定类型的文件。在Python中，我们可以使用字符串方法来实现这一操作。我们首先使用os.path.splitext()函数来分离文件路径和文件名后缀。然后，我们可以使用字符串方法来检查文件名后缀是否符合要求。

import os

def list_files_by_extension(path, extension):
    # 递归地列出目录下指定后缀名的所有文件
    for root, dirs, files in os.walk(path):
        for filename in files:
            # 获取文件的完整路径
            file_path = os.path.join(root, filename)
            # 分离文件名和文件名后缀，获取文件名后缀
            _, file_extension = os.path.splitext(file_path)
            # 判断文件名后缀是否匹配
            if file_extension == extension:
                print(file_path) # 输出文件的完整路径

# 获取当前工作目录
cwd = os.getcwd()

# 列出目录下的所有.png文件
list_files_by_extension(cwd, ".png")

上述代码中，我们定义了一个名为list_files_by_extension()的函数，该函数可以筛选出指定的文件类型。我们首先使用os.path.splitext()函数获取文件的后缀名，然后判断该后缀名是否匹配我们想要的后缀名，并输出完整文件路径。

五、结论

在本文中，我们深入探讨了如何使用Python列出目录下的所有文件。从获取当前工作目录开始，我们通过os模块提供的函数和方法，介绍了多个列出文件的方法。此外，我们还介绍了如何递归地列出目录下的所有文件，以及如何根据文件名后缀来筛选指定类型的文件。这些方法将在文件操作中提供帮助。