matplotlib绘图如何将绘出的图像保存为变量

您可以使用`matplotlib.pyplot.gcf()`函数获取当前绘图的Figure对象，然后使用`Figure.canvas`属性将图像保存为一个numpy数组。示例如下：

import matplotlib.pyplot as plt
import numpy as np

# 绘图
x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)
plt.plot(x, y)

# 将图像保存为numpy数组
fig = plt.gcf() # 获取当前绘图的Figure对象
fig.canvas.draw() # 绘制图形
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))

# 显示图像
plt.show()

# data即为保存的numpy数组

您也可以使用`Figure.savefig()`函数将图像保存到文件中，示例如下：

import matplotlib.pyplot as plt
import numpy as np

# 绘图
x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)
plt.plot(x, y)

# 将图像保存到文件
fig = plt.gcf() # 获取当前绘图的Figure对象
fig.savefig('my_figure.png') # 保存图像到文件

# 显示图像
plt.show()

以上代码将生成一个名为`my_figure.png`的文件，其中包含绘制的图像。