matplotlib.pyplot画图 图片的二进制流的获取方法

通过使用matplotlib模块的子模块pyplot可以方便地进行数据可视化和绘图。在这个过程中，有时需要将图片作为二进制流的形式获取，以便于后续使用，本篇文章将详细讲解如何获取图片的二进制流。

1. 获取画图对象

在使用pyplot绘图时，我们需要先创建一个画图对象，比如下面的代码：

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
plt.show()

这段代码将创建一个简单的折线图，并通过show()函数显示在屏幕上。

2. 获取二进制流

要获取这个图像的二进制流，我们可以使用io模块中的BytesIO类。下面的代码示例中，我们将把图像保存为PNG格式，并在内存中获取其二进制流形式：

import io
import matplotlib.pyplot as plt

fig = plt.figure()

# 绘制图形
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])

# 保存图像
buffer = io.BytesIO()
fig.savefig(buffer, format='png')

# 从内存中获取二进制流
buffer.seek(0)
image_binary = buffer.getvalue()

在这个代码过程中，我们创建了一个BytesIO对象，并将其作为参数传递给savefig()函数，以将图像保存为PNG格式，并将其存储在内存中。然后，我们使用seek()函数将位置指针移动到内存的开头，最后使用getvalue()函数从内存中获取二进制流。

3. 显示图像

为了验证我们的代码是否工作正常，我们可以使用下面的代码将图像从二进制流中显示出来：

from PIL import Image

# 从二进制流中创建图像对象
img = Image.open(io.BytesIO(image_binary))

# 显示图像
img.show()

这段代码将创建一个Pillow图像对象，并使用show()函数在屏幕上显示出来。

示例

以下是一个完整的示例代码，将创建一个简单的折线图，并将其保存为PNG格式的二进制流：

import io
import matplotlib.pyplot as plt
from PIL import Image


# 绘制图形
def plot_image():
    fig = plt.figure()
    plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
    buffer = io.BytesIO()
    fig.savefig(buffer, format='png')
    buffer.seek(0)
    image_binary = buffer.getvalue()
    img = Image.open(io.BytesIO(image_binary))
    img.show()


if __name__ == "__main__":
    plot_image()

可以看出，这段代码将创建一个简单的折线图，并使用BytesIO保存其为PNG格式二进制流，最后将其转化为Pillow图像对象并显示在屏幕上。

当然，可以根据需要对这个代码进行更改：比如更换图形类型、更改图形大小等。但总的来说，这些示例代码应该能够为你提供足够的灵感，帮助你在python中更方便的处理图像。