python如何获取文件夹下的文件名

Python 获取文件夹下的文件名的方法有很多，包括使用os模块、glob模块和pathlib模块等。 在这篇博客文章中，我们将详细介绍这些方法，并给出示例代码。最推荐的方法是使用os模块，因为它提供了简单易用的函数，可以方便地遍历文件系统。

一、使用 os 模块获取文件夹下的文件名

os模块是Python标准库的一部分，提供了与操作系统进行交互的功能。使用os模块可以非常方便地获取文件夹下的文件名。

1. os.listdir() 方法

os.listdir() 函数返回指定文件夹中的所有文件和文件夹的名称列表。

import os
def list_files(directory):
    return os.listdir(directory)
示例
directory_path = '/path/to/your/directory'
files = list_files(directory_path)
print(files)

2. os.walk() 方法

os.walk()函数生成文件名，可以遍历目录树。

import os
def list_files_walk(directory):
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            print(os.path.join(dirpath, filename))
示例
directory_path = '/path/to/your/directory'
list_files_walk(directory_path)

二、使用 glob 模块获取文件夹下的文件名

glob模块提供了一个功能来查找符合特定模式的文件路径名。

1. glob.glob() 方法

glob.glob() 函数返回所有匹配的文件路径列表。

import glob
def list_files_glob(directory):
    return glob.glob(directory + '/*')
示例
directory_path = '/path/to/your/directory'
files = list_files_glob(directory_path)
print(files)

三、使用 pathlib 模块获取文件夹下的文件名

pathlib模块是Python 3.4引入的新模块，提供了面向对象的文件系统路径操作方法。

1. Path().iterdir() 方法

Path().iterdir() 函数返回一个生成器，用于遍历目录中的所有文件和文件夹。

from pathlib import Path
def list_files_pathlib(directory):
    path = Path(directory)
    return [str(file) for file in path.iterdir() if file.is_file()]
示例
directory_path = '/path/to/your/directory'
files = list_files_pathlib(directory_path)
print(files)

四、比较不同方法的优缺点

1. os模块的优缺点

优点：

简单易用，适合初学者。
功能强大，可以进行复杂的文件和目录操作。

缺点：

代码较为冗长。
不支持文件路径的模式匹配。

2. glob模块的优缺点

优点：

支持文件路径的模式匹配，非常适合查找特定类型的文件。
代码简洁。

缺点：

只能匹配文件名，不适合复杂的文件和目录操作。

3. pathlib模块的优缺点

优点：

面向对象，代码简洁。
功能强大，可以进行复杂的文件和目录操作。
支持文件路径的模式匹配。

缺点：

需要Python 3.4及以上版本。

五、实战示例：获取特定类型文件

有时候我们需要获取特定类型的文件，比如所有的.txt文件。我们可以结合上述方法实现这一需求。

1. 使用 os模块和字符串方法

import os
def list_txt_files(directory):
    return [file for file in os.listdir(directory) if file.endswith('.txt')]
示例
directory_path = '/path/to/your/directory'
txt_files = list_txt_files(directory_path)
print(txt_files)

2. 使用 glob模块

import glob
def list_txt_files_glob(directory):
    return glob.glob(directory + '/*.txt')
示例
directory_path = '/path/to/your/directory'
txt_files = list_txt_files_glob(directory_path)
print(txt_files)

3. 使用 pathlib模块

from pathlib import Path
def list_txt_files_pathlib(directory):
    path = Path(directory)
    return [str(file) for file in path.glob('*.txt') if file.is_file()]
示例
directory_path = '/path/to/your/directory'
txt_files = list_txt_files_pathlib(directory_path)
print(txt_files)

六、处理子目录中的文件

有时候我们需要遍历子目录中的文件，这时可以使用os.walk()方法或pathlib模块中的rglob()方法。

1. 使用 os.walk() 方法

import os
def list_all_files(directory):
    files_list = []
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            files_list.append(os.path.join(dirpath, filename))
    return files_list
示例
directory_path = '/path/to/your/directory'
all_files = list_all_files(directory_path)
print(all_files)

2. 使用 pathlib 模块

from pathlib import Path
def list_all_files_pathlib(directory):
    path = Path(directory)
    return [str(file) for file in path.rglob('*') if file.is_file()]
示例
directory_path = '/path/to/your/directory'
all_files = list_all_files_pathlib(directory_path)
print(all_files)