Python中的图片格式判断：imghdr模块使用指南

在Python中，可以使用imghdr模块来判断图片的格式。imghdr模块提供了一个简单的方法，可以根据文件的内容来判断文件的格式是否为图片格式。以下是imghdr模块的使用指南，并附带使用例子。

1. 导入imghdr模块

首先，我们需要导入imghdr模块，使用以下代码：

   import imghdr
   

2. 使用imghdr模块的方法

imghdr模块提供了一个方法imghdr.what(filename)，该方法接受一个文件名作为参数，并返回文件的格式。具体的使用方法如下：

   image_format = imghdr.what(filename)
   print("Image format: ", image_format)
   

在上述代码中，filename是要判断的图片文件的路径。imghdr.what方法会返回一个字符串，表示图片的格式，例如jpeg、png、gif等。如果文件不是图片格式，则返回None。

注意：imghdr.what方法只读取文件的内容进行判断，并不会检查文件扩展名。

3. 完整的使用例子

下面是一个完整的使用imghdr模块判断图片格式的例子：

   import imghdr

   def check_image_format(filename):
       image_format = imghdr.what(filename)
       if image_format is not None:
           print("Image format: ", image_format)
       else:
           print("Not a valid image file.")

   check_image_format("image.jpg")
   check_image_format("image.png")
   check_image_format("image.gif")
   check_image_format("document.pdf")
   

在上述例子中，我们定义了一个check_image_format函数，用来判断图片文件的格式。首先，我们调用imghdr.what方法获取图片文件的格式，然后根据结果输出相应的信息。

示例中，我们测试了一个jpg图片、一个png图片和一个gif图片，分别得到了对应的图片格式。而对于一个pdf文件，由于不是图片格式，所以输出了"Not a valid image file."的信息。

通过使用imghdr模块，我们可以很方便地判断图片文件的格式，从而进行相应的处理。