如何将折线图转为像素强度二维数组并用imshow()绘制？

你可以直接利用matplotlib的画布缓冲区获取像素数据，无需保存为PNG文件，具体步骤如下：

核心实现代码 #

import numpy as np
import matplotlib.pyplot as plt

# 创建无框画布并绘制折线
fig = plt.figure(frameon=False)
plt.plot(np.arange(5), [4,5,2,7,4], color="k", alpha=0.3)
plt.plot(np.arange(5), [4,6,2,8,4], color="k", alpha=0.3)
plt.axis('off')

# 渲染画布到内存缓冲区
fig.canvas.draw()

# 从缓冲区提取像素数据并转换为二维数组
# 获取ARGB格式的像素字节流，转为numpy数组
pixel_buffer = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
# 调整形状为 (高度, 宽度, 4)，并将ARGB转为RGBA顺序
height, width = fig.canvas.get_height_width()[::-1]
rgba_array = pixel_buffer.reshape(height, width, 4)[:, :, [1,2,3,0]]

# 提取像素强度数组（这里取alpha通道，对应折线的透明度强度）
intensity_array = rgba_array[:, :, 3]

# 用imshow重现图像
plt.figure()
plt.imshow(intensity_array, cmap='gray')
plt.axis('off')
plt.show()

关键步骤说明 #

• fig.canvas.draw()：强制matplotlib将画布内容渲染到内存缓冲区，这是获取像素数据的前提
• fig.canvas.tostring_argb()：导出缓冲区中的像素数据，格式为ARGB（Alpha-Red-Green-Blue）的字节流
• 数组形状调整：将一维字节数组转为二维图像数组，并调整通道顺序为常用的RGBA
• 强度提取：由于原折线是黑色（RGB通道值为0）带透明度，直接取Alpha通道即可得到像素强度（0为透明背景，76对应alpha=0.3的255*0.3≈76）；如果是彩色折线，可以通过RGB分量计算亮度：intensity_array = 0.299*rgba_array[:,:,0] + 0.587*rgba_array[:,:,1] + 0.114*rgba_array[:,:,2]

后续更新图像的思路 #

要添加新折线时，无需重新绘制所有历史折线：

1. 创建临时画布绘制新折线，按同样方法提取其强度数组
2. 将新的强度数组与原intensity_array叠加（注意用np.clip限制值在0-255范围内，避免溢出）
3. 直接用plt.imshow()渲染更新后的数组即可

内容的提问来源于stack exchange，提问作者Tomoon