如何在Python中存储Matplotlib绘图以便后续调用至VTK对象？

首先得指出你原来代码的核心问题：plt.show()会显示图形并清空当前Matplotlib画布，而且它的返回值是None——所以你的plot_l列表里其实全是无效的空值，这肯定达不到你要的效果。

要实现内存中存储多个绘图、可索引调用的需求，有两种推荐方案，取决于你后续的使用场景：

方案1：直接存储Matplotlib Figure对象（推荐，灵活度高） #

Matplotlib的Figure是整个绘图的核心对象，你可以直接把每个绘图的Figure实例存入列表，后续不仅能随时渲染成PNG，还能修改图形内容。

示例代码： #

import matplotlib.pyplot as plt

plot_l = []
# 假设x是你的数据列表，theta和r是每个绘图的极坐标数据
for i in range(len(x)):
    # 为每个绘图创建独立的Figure和极坐标Axes
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='polar')
    
    # 这里替换成你的绘图逻辑，比如：
    theta = ...  # 你的极角数据
    r = ...      # 你的极径数据
    ax.plot(theta, r, label=f"Plot {i}")
    ax.legend()
    
    # 将Figure对象存入列表
    plot_l.append(fig)
    
    # 不要调用plt.show()！除非你需要实时显示，否则会清空画布

调用方式： #

当你需要使用第10个绘图时，直接取plot_l[10]，这个对象包含了完整的绘图信息。后续如果要转换成PNG字节流（用于VTK），可以这样做：

from io import BytesIO

fig = plot_l[10]
buf = BytesIO()
fig.savefig(buf, format='png', dpi=100)  # dpi可按需调整
buf.seek(0)  # 将指针重置到字节流开头，方便读取

方案2：存储内存中的PNG字节流（内存占用更低） #

如果你的绘图数量极大，且不需要后续修改图形，可以直接把每个绘图渲染成PNG字节流存入列表，这样内存占用会比存储Figure对象小很多。

示例代码： #

import matplotlib.pyplot as plt
from io import BytesIO

plot_l = []
for i in range(len(x)):
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='polar')
    
    # 你的绘图逻辑
    ax.plot(theta, r, label=f"Plot {i}")
    ax.legend()
    
    # 将Figure保存到内存中的BytesIO对象
    buf = BytesIO()
    fig.savefig(buf, format='png')
    buf.seek(0)
    
    # 存入列表
    plot_l.append(buf)
    
    # 关闭Figure释放内存（因为已经存了字节流，不需要保留Figure了）
    plt.close(fig)

调用方式： #

取第10个绘图的PNG字节：

png_bytes = plot_l[10].getvalue()

如何将这些数据导入VTK？ #

不管用哪种方案，最终都可以把PNG数据转换成VTK的图像对象，比如用vtkImageData：

import vtk
import vtk.util.numpy_support as ns
from PIL import Image
import numpy as np

def convert_png_to_vtk_image(png_bytes):
    # 用PIL读取PNG字节
    img = Image.open(BytesIO(png_bytes))
    img_np = np.array(img)
    
    # 将numpy数组转换成VTK图像数据
    vtk_array = ns.numpy_to_vtk(
        num_array=img_np.flatten(),
        deep=True,
        array_type=vtk.VTK_UNSIGNED_CHAR
    )
    vtk_array.SetNumberOfComponents(4)  # 假设是RGBA格式的PNG
    
    image_data = vtk.vtkImageData()
    image_data.SetDimensions(img.width, img.height, 1)
    image_data.GetPointData().SetScalars(vtk_array)
    
    return image_data

# 调用示例：
vtk_img = convert_png_to_vtk_image(plot_l[10].getvalue())

这样你就可以把vtk_img作为纹理或图像元素添加到VTK场景中了。

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