如何使用imghdr模块判断图片文件的类型

imghdr模块是Python标准库中的一个模块，用于判断图片文件的类型。它提供了一个简单的接口，可以根据文件的内容判断图片的类型。

使用imghdr模块非常简单，只需要调用imghdr.what()函数就可以判断图片文件的类型。下面是使用imghdr模块判断图片文件类型的例子：

import imghdr

# 例子1：判断单个图片文件的类型
filename = 'example.jpg'
file_type = imghdr.what(filename)
print('图片文件类型:', file_type)

# 例子2：批量判断文件夹中的图片文件类型
import os

def get_image_files(folder):
    image_files = []
    for root, dirs, files in os.walk(folder):
        for file in files:
            file_path = os.path.join(root, file)
            file_type = imghdr.what(file_path)
            if file_type:
                image_files.append((file_path, file_type))
    return image_files

folder = 'images'
image_files = get_image_files(folder)
print('图片文件列表:')
for file_path, file_type in image_files:
    print(f'文件路径: {file_path}\t\t类型: {file_type}')

上述例子中，首先导入了imghdr模块。然后，在例子1中，我们使用imghdr.what()函数判断了一个图片文件的类型，函数的参数是图片文件的路径，函数的返回值是一个字符串，表示图片的类型。通过打印文件类型，我们可以得到图片文件的类型。

在例子2中，我们定义了一个get_image_files()函数，用于批量获取文件夹中的图片文件。函数的参数是文件夹的路径，函数的返回值是一个列表，列表的每个元素是一个元组，包含图片文件的路径和文件类型。在函数中，我们使用os模块的walk()函数来遍历文件夹中的文件，然后使用imghdr.what()函数判断每个文件的类型，如果返回值不为空，则将文件路径和文件类型添加到列表中。最后，通过遍历列表，我们可以打印出文件路径和文件类型。

需要注意的是，imghdr模块只能判断图片文件的类型，不能判断图片文件的后缀名。它通过读取文件内容的方式判断文件类型，所以对于一些特殊格式的图片文件可能无法准确判断类型。另外，imghdr模块只能判断图片文件的类型，不能用于其他类型的文件。

总结来说，使用imghdr模块判断图片文件类型非常简单，只需要调用imghdr.what()函数即可。通过传入文件路径作为函数的参数，可以得到文件的类型。在实际使用中，可以根据文件类型进行相应的处理或者过滤。