Python 中os模块/pathlib模块获取文件夹下的文件名

使用Python的os模块或pathlib模块来获取文件夹下的所有文件名。以下是两种方法的示例代码：

方法1：使用os模块

import os

def get_file_names(directory):
    # 获取目录下的所有文件和文件夹
    all_items = os.listdir(directory)
    
    # 过滤出文件
    file_names = [item for item in all_items if os.path.isfile(os.path.join(directory, item))]
    
    return file_names

# 示例用法
directory_path = '/path/to/your/directory'
file_names = get_file_names(directory_path)
print(file_names)

方法2：使用pathlib模块

from pathlib import Path

def get_file_names(directory):
    # 使用Path对象
    path = Path(directory)
    
    # 获取目录下的所有文件
    file_names = [file.name for file in path.iterdir() if file.is_file()]
    
    return file_names

# 示例用法
directory_path = '/path/to/your/directory'
file_names = get_file_names(directory_path)
print(file_names)

解释：

• os.listdir(directory)：返回指定目录下的所有文件和文件夹的名称列表。
• os.path.isfile(path)：检查给定的路径是否是一个文件。
• Path(directory).iterdir()：返回一个生成器，生成目录下的所有文件和文件夹的Path对象。
• file.is_file()：检查Path对象是否是一个文件。

注意事项：

• 确保提供的directory_path是一个有效的目录路径。
• 如果你需要递归地获取子目录中的文件，可以使用os.walk()或Path.rglob()。