Python使用imghdr模块检测图片类型

向右看齐 2024-03-22 13:37 48阅读 0赞

在Python中，我们经常需要处理图片文件。但是，有时候我们并不知道图片的具体类型，这时候就需要使用imghdr模块来检测图片类型。

imghdr模块是Python自带的模块，可以用来检测图片文件的类型。它可以检测常见的图片格式，如JPEG、PNG、GIF等。

下面是一个简单的示例代码，演示如何使用imghdr模块检测图片类型：

import imghdr
    
    # 定义一个图片文件路径
    img_path = 'test.jpg'
    
    # 使用imghdr模块检测图片类型
    img_type = imghdr.what(img_path)
    
    # 输出图片类型
    print('图片类型为：', img_type)

在上面的代码中，我们首先导入了imghdr模块。然后，我们定义了一个图片文件路径，这里我们使用了一个名为test.jpg的图片文件。接着，我们使用imghdr模块的what()函数来检测图片类型，并将结果赋值给img\_type变量。最后，我们输出了图片类型。

运行上面的代码，输出结果如下：

图片类型为： jpeg

可以看到，我们成功地检测出了图片的类型，它是JPEG格式的。

除了检测单个图片文件的类型，imghdr模块还可以检测一组图片文件的类型。下面是一个示例代码，演示如何检测一组图片文件的类型：

import imghdr
    
    # 定义一组图片文件路径
    img_paths = ['test.jpg', 'test.png', 'test.gif']
    
    # 使用imghdr模块检测图片类型
    img_types = [imghdr.what(path) for path in img_paths]
    
    # 输出图片类型
    print('图片类型为：', img_types)

在上面的代码中，我们定义了一个包含三个图片文件路径的列表。然后，我们使用imghdr模块的what()函数来检测每个图片文件的类型，并将结果存储在一个列表中。最后，我们输出了所有图片文件的类型。

运行上面的代码，输出结果如下：

图片类型为： ['jpeg', 'png', 'gif']

可以看到，我们成功地检测出了三个图片文件的类型，它们分别是JPEG、PNG和GIF格式的。

总结：

imghdr模块是Python自带的模块，可以用来检测图片文件的类型。它可以检测常见的图片格式，如JPEG、PNG、GIF等。使用imghdr模块非常简单，只需要调用what()函数即可。如果需要检测一组图片文件的类型，可以使用列表推导式来实现。