Pythonimghdr模块：轻松判断图像文件类型

在Python中，图像处理是一个很常见的任务。在处理图像之前，通常我们需要知道图像的文件类型，以便选择不同的处理方法。Python的imghdr模块提供了一种简单的方法来判断图像文件的类型。

imghdr模块中的主要函数是imghdr.what()，它可以根据文件的内容来判断文件类型。该函数接受一个文件名作为参数，并返回文件类型的字符串。以下是imghdr.what()函数支持的图像文件类型列表：

- 'bmp'：Windows位图文件

- 'gif'：GIF格式图像

- 'jpeg'或'jpg'：JPEG格式图像

- 'png'：PNG格式图像

- 'tiff'或'tif'：TIFF格式图像

- 'rast'或'ras'：Sun光栅图

- 'xbm'：X11位图文件

- 'pnm'：PBM, PGM, PPM或PAM格式图像

下面是一个使用imghdr.what()函数的例子：

import imghdr

def check_image_type(file_name):
    image_type = imghdr.what(file_name)
    if image_type:
        print(f'The file {file_name} is of type {image_type}')
    else:
        print(f'Cannot determine the file type of {file_name}')

check_image_type('image.jpg')
check_image_type('image.png')
check_image_type('image.gif')

输出：

The file image.jpg is of type jpeg
The file image.png is of type png
The file image.gif is of type gif

在上面的代码中，我们定义了一个check_image_type()函数，它接受一个文件名作为参数。该函数首先调用imghdr.what()函数来判断文件的类型，并将返回的文件类型打印出来。如果无法确定文件类型，将打印出相应的提示信息。

在这个例子中，我们分别检查了一个JPEG格式的图像文件、一个PNG格式的图像文件和一个GIF格式的图像文件。根据输出结果，我们可以确认这些文件的类型。

总结一下，imghdr模块为我们提供了一个简单方便的方法来判断图像文件的类型。通过调用imghdr.what()函数，我们可以轻松确定图像文件的类型，并相应地进行处理。