在 python pillow 中直接显示 matplotlib 生成的图片

在本文中，我们将介绍如何在 pillow 库中直接显示 matplotlib 生成的图片，而无需创建中间文件。

通常，在使用 pillow 显示 matplotlib 生成的图片时，我们需要先保存图片到文件中，然后再使用 pillow 打开。这种方式对于需要保存图片到磁盘的情况是合适的，但当我们只想在内存中显示图片时，就会造成不必要的开销。

Python Pep Authoring

依据官方 python/peps仓库规则起草、审阅、编辑或准备 Python 改进提案（PEP），用于处理 PEP 文本和元数据头部。

下载

为了避免创建中间文件，我们可以使用 io.bytesio 类来创建一个类似于文件的对象，该对象可以存储图片数据。这种方法被称为“鸭子类型”，因为它允许我们使用 pillow 打开 bytesio 对象，即使它不是真正的文件。

具体步骤如下：

1. 使用 matplotlib 生成图片并将其保存到 bytesio 对象中：

import io
import matplotlib.pyplot as plt
from pil import image

# 生成图片
plt.plot([1, 2, 3, 4])

# 保存图片到 bytesio 对象
output = io.bytesio()
plt.savefig(output, format="png")

2. 使用 pillow 打开 bytesio 对象中的图片并显示：

# 从 BytesIO 对象中读取图片数据
pil_image = Image.open(output)

# 显示图片
pil_image.show()

通过这种方法，我们可以在不创建中间文件的情况下直接在 pillow 中显示 matplotlib 生成的图片。它对于需要快速显示图片或在内存中处理图片的应用程序很有用。