python - 如何为 matplotlib 图分配变量名？

Question

我正在尝试为生成的 matplotlib 图片分配一个特定的变量名称。但是，我无法这样做。这是代码：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib
import numpy as np
from matplotlib import cm
from matplotlib import pyplot as plt
step = 0.04
maxval = 1.0
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

matplotlib.pyplot.jet()
r = np.linspace(0,1.25,50)
p = np.linspace(0,2*np.pi,50)
R,P = np.meshgrid(r,p)
X,Y = R*np.cos(P),R*np.sin(P)

Z = ((R**2 - 1)**2)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet)
ax.set_zlim3d(0, 1)
ax.set_xlabel(r'$\phi_\mathrm{real}$')
ax.set_ylabel(r'$\phi_\mathrm{im}$')
ax.set_zlabel(r'$V(\phi)$')

kurs = 252

plt.savefig(kurs, format='png')

如您所见，我正在尝试分配一个名为“kurs”的变量作为图片的名称。任何帮助将不胜感激。谢谢你。

Answer 1

文件名必须是字符串，而不是数字。尝试：

kurs = '252'

仍然使用'252.png'行会更方便。
如果您从某种数字生成器中分配名称，您可以这样做：

for i in range(10):
    ...................
    ...................
    kurs = "%i.png" % i
    plt.savefig(kurs, format='png')