python - plt.save 保存不同的图像

Question

我有以下代码:

plt.figure()
plt.plot(Xu,Yu, 'g>',label='u')
plt.plot(Xv,Yv,'r^',label='v')
plt.plot(Xc,Yc,'kx',label='p')
plt.xlim(-0.2,1.2)
plt.ylim(-0.2,1.2)
plt.scatter(X,Y)
plt.grid()
#plt.legend()
plt.savefig('Grid.png')

这使得下图:

但是，当我在保存的文件夹中打开它时，图例消失了。

注意:如果我取消注释图例行，结果是相同的，保存的图像如下所示:

我该怎么做才能准确保存我创建的图像而不是更改的版本？

Answer 1

你们的速度和压力阵列的形状是什么？如果您将它们作为行和列的二维数组，那么 matplotlib 将假定每一行都是一个单独的图。作为解决方案，您可以将它们扩展为一维数组，

plt.plot(Xu.ravel(),Yu.ravel(), 'g>',label='u')
plt.plot(Xv.ravel(),Yv.ravel(),'r^',label='v')
plt.plot(Xc.ravel(),Yc.ravel(),'kx',label='p')

如果这不起作用，您可以更明确地标记前三个速度和压力位置并抑制其余位置，

#Plot first element with labels
plt.plot(Xu.ravel()[0],Yu.ravel()[0], 'g>',label='u')
plt.plot(Xv.ravel()[0],Yv.ravel()[0],'r^',label='v')
plt.plot(Xc.ravel()[0],Yc.ravel()[0],'kx',label='p')

#plot remaining without Legend
plt.plot(Xu.ravel()[1:],Yu.ravel()[1:], 'g>',legend=False)
plt.plot(Xv.ravel()[1:],Yv.ravel()[1:],'r^',legend=False)
plt.plot(Xc.ravel()[1:],Yc.ravel()[1:],'kx',legend=False)