097、Python 获取文件夹下的所有文件的方法

在Python中，获取一个文件夹下的所有文件是一个常见的操作，特别是在处理文件管理、大数据分析等任务时。下面我们将介绍几种获取文件夹下所有文件的方法，并通过代码示例来说明它们的使用。

方法一：使用os模块

os模块是Python标准库中一个非常实用的模块，提供了与操作系统交互的方法，可以用来访问和操作文件和目录。

import os

def list_files_in_directory(directory):
    try:
        # 获取指定目录下的所有文件和文件夹
        items = os.listdir(directory)
        files = [item for item in items if os.path.isfile(os.path.join(directory, item))]
        return files
    except Exception as e:
        print(f"Error occurred: {e}")
        return []

# 示例用法
directory_path = "你的文件夹路径"
files = list_files_in_directory(directory_path)
print("目录下的所有文件：", files)

在这个示例中，我们首先导入了os模块，然后定义了一个函数list_files_in_directory(directory)，接受一个文件夹路径作为参数。使用os.listdir()方法列出文件夹下的所有项，并通过os.path.isfile()来检查每一项是否为文件，最终返回文件名列表。

方法二：使用os.walk()

os.walk()是一个更强大的函数，可以递归遍历文件夹及其子文件夹，获取所有层级的文件。

import os

def list_all_files_in_directory(directory):
    all_files = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            all_files.append(os.path.join(root, file))  # 获取完整路径
    return all_files

# 示例用法
directory_path = "你的文件夹路径"
all_files = list_all_files_in_directory(directory_path)
print("所有文件（包含子文件夹中的文件）：", all_files)

在这个例子中，os.walk()会返回一个生成器，其中每次迭代会返回一个包含当前目录路径、子目录名称和文件名的元组。通过嵌套循环，我们可以将所有文件的完整路径添加到列表中。

方法三：使用glob模块

glob模块也可以用来查找指定类型的文件。它支持Unix风格的路径名模式扩展。

import glob
import os

def list_specific_files(directory, extension="*"):
    # 使用glob.glob()列出特定扩展名的文件
    search_pattern = os.path.join(directory, f"*.{extension}")
    files = glob.glob(search_pattern)
    return files

# 示例用法
directory_path = "你的文件夹路径"
files = list_specific_files(directory_path, "txt")  # 获取所有txt文件
print("所有TXT文件：", files)

在这个示例中，glob.glob()用于查找与给定模式匹配的所有路径名，os.path.join()确保路径格式正确。通过指定文件扩展名，可以过滤出特定类型的文件。

总结

我们介绍了三种获取文件夹下所有文件的方法：使用os模块、os.walk()函数和glob模块。根据不同的需求，可以选择最适合的方法。如果你想获取所有文件及其子文件夹中的文件，os.walk()是推荐的方式。如果只想列出特定类型的文件，使用glob模块会更加方便。每种方法都简单易用，是Python文件处理的基础。