获取文件夹python中的所有类型的图像

当我们需要处理一个文件夹里的所有图像文件时，如何快速地获取这些图像呢？以下是Python代码实现该功能的方式。

import os

def get_images(folder_path:str):
    """
    获取指定文件夹内的所有图像文件路径
    :param folder_path: 文件夹路径
    :return: 图像文件路径列表
    """
    images = []
    for filename in os.listdir(folder_path):
        # 判断文件是否是图像文件
        if filename.endswith(".jpg") or filename.endswith(".jpeg") or filename.endswith(".png") or filename.endswith(".bmp") or filename.endswith(".gif"):
            images.append(os.path.join(folder_path, filename))
    return images

以上代码使用了Python自带的os模块，通过os.listdir()函数遍历文件夹内的所有文件名，并且筛选出图像文件添加到列表中。

如果需要获取文件夹中所有类型的文件，我们可以使用os.path.splitext()函数来获取文件后缀名，然后做出相应的判断。以下是更加通用的代码：

import os

def get_files(folder_path:str):
    """
    获取指定文件夹内的所有文件路径
    :param folder_path: 文件夹路径
    :return: 文件路径列表
    """
    files = []
    for filename in os.listdir(folder_path):
        # 获取文件后缀名
        _, extension = os.path.splitext(filename)
        # 判断文件是否是图像文件
        if extension.lower() in [".jpg", ".jpeg", ".png", ".bmp", ".gif"]:
            files.append(os.path.join(folder_path, filename))
    return files

以上代码将所有文件的后缀名与包含图像文件后缀名的列表进行比较，若相同则添加该文件路径到列表中。

通过以上代码实现，我们可以方便地获取指定文件夹内的所有图像或所有类型的文件。