Python怎么获取指定目录下的所有文件

在Python中，获取指定目录下的所有文件是一个常见的任务。无论是为了文件管理、数据分析还是自动化脚本，掌握如何遍历目录并获取文件列表都是非常有用的。本文将介绍几种常用的方法来实现这一目标。

1. 使用os模块

os模块是Python标准库中的一个核心模块，提供了许多与操作系统交互的函数。其中，os.listdir()函数可以列出指定目录下的所有文件和子目录。

import os

def get_files_in_directory(directory):
    files = []
    for item in os.listdir(directory):
        item_path = os.path.join(directory, item)
        if os.path.isfile(item_path):
            files.append(item_path)
    return files

# 示例用法
directory = '/path/to/your/directory'
files = get_files_in_directory(directory)
print(files)

优点：

• 简单易用，适合快速获取文件列表。
• 不需要额外的依赖。

缺点：

• 只能获取当前目录下的文件，不能递归获取子目录中的文件。

2. 使用os.walk()递归获取文件

如果你需要递归地获取目录及其子目录中的所有文件，可以使用os.walk()函数。os.walk()会生成一个三元组(dirpath, dirnames, filenames)，其中filenames是当前目录下的文件列表。

import os

def get_all_files(directory):
    all_files = []
    for dirpath, _, filenames in os.walk(directory):
        for filename in filenames:
            all_files.append(os.path.join(dirpath, filename))
    return all_files

# 示例用法
directory = '/path/to/your/directory'
files = get_all_files(directory)
print(files)

优点：

• 可以递归获取目录及其子目录中的所有文件。
• 灵活性高，可以根据需要处理目录和文件。

缺点：

• 对于非常大的目录结构，可能会占用较多内存。

3. 使用glob模块

glob模块提供了一种基于通配符的文件路径匹配方式。你可以使用glob.glob()函数来获取匹配特定模式的文件列表。

import glob

def get_files_by_pattern(directory, pattern='*'):
    return glob.glob(os.path.join(directory, pattern))

# 示例用法
directory = '/path/to/your/directory'
files = get_files_by_pattern(directory, '*.txt')  # 获取所有.txt文件
print(files)

优点：

• 支持通配符匹配，适合需要筛选特定类型文件的场景。
• 简单易用。

缺点：

• 不支持递归匹配子目录中的文件（除非使用**通配符，但需要Python 3.5+）。

4. 使用pathlib模块

pathlib是Python 3.4引入的一个模块，提供了面向对象的路径操作方式。使用Path对象的rglob()方法可以递归获取目录下的所有文件。

from pathlib import Path

def get_all_files_with_pathlib(directory):
    path = Path(directory)
    return [str(file) for file in path.rglob('*') if file.is_file()]

# 示例用法
directory = '/path/to/your/directory'
files = get_all_files_with_pathlib(directory)
print(files)

优点：

• 面向对象的设计，代码更易读。
• 支持递归获取文件，且语法简洁。

缺点：

• 需要Python 3.4及以上版本。

5. 使用scandir模块（Python 3.5+）

scandir模块是os模块的一个扩展，提供了更高效的文件系统遍历方式。os.scandir()函数返回一个DirEntry对象的迭代器，可以更快地获取文件信息。

import os

def get_files_with_scandir(directory):
    files = []
    with os.scandir(directory) as it:
        for entry in it:
            if entry.is_file():
                files.append(entry.path)
    return files

# 示例用法
directory = '/path/to/your/directory'
files = get_files_with_scandir(directory)
print(files)

优点：

• 性能优于os.listdir()，特别是在处理大量文件时。
• 提供了更多的文件信息（如文件类型、大小等）。

缺点：

• 需要Python 3.5及以上版本。

总结

在Python中，获取指定目录下的所有文件有多种方法，每种方法都有其适用的场景。如果你只需要获取当前目录下的文件，os.listdir()是一个简单直接的选择。如果需要递归获取子目录中的文件，os.walk()或pathlib的rglob()方法更为合适。对于需要筛选特定类型文件的场景，glob模块提供了便捷的通配符匹配功能。而scandir模块则在处理大量文件时提供了更好的性能。

根据你的具体需求，选择合适的方法可以让你更高效地完成任务。