python：获取某路径下所有图片的名称

可以使用 Python 的 os 模块或者 pathlib 模块来获取指定路径下所有图片的名称。以下是使用这两种方法实现的代码示例：

使用 os 模块

import os

def get_image_names_os(path):
    image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp')
    image_names = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if file.lower().endswith(image_extensions):
                image_names.append(file)
    return image_names

# 示例用法
path = '.'  # 当前目录
image_names = get_image_names_os(path)
print(image_names)

使用 pathlib 模块

from pathlib import Path

def get_image_names_pathlib(path):
    image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp')
    path = Path(path)
    image_names = [p.name for p in path.rglob('*') if p.is_file() and p.suffix.lower() in image_extensions]
    return image_names

# 示例用法
path = '.'  # 当前目录
image_names = get_image_names_pathlib(path)
print(image_names)

上述代码定义了两个函数 get_image_names_os 和 get_image_names_pathlib 分别使用 os 模块和 pathlib 模块来获取指定路径下所有图片的名称。函数接受一个路径作为参数，返回一个包含所有图片名称的列表。你可以根据需要修改 path 变量的值来指定不同的路径。