Python怎么獲取指定目錄下的所有文件

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模塊則在處理大量文件時提供了更好的性能。

根據你的具體需求，選擇合適的方法可以讓你更高效地完成任務。