Matplotlib - 不同时刻特定轮廓的图例

Question

我试图在不同的时间绘制具有特定轮廓线（水平= 320）的图形，这就是使用循环的原因。 我想用标签绘制一个图例来指定循环中的时间，如下所示：

我的代码的一部分显示：

cmap = plt.cm.hot

instant = 0
for instant in range(0,Sigma_stockMAX.shape[2]):

    name = 'test'

    VM_MAX_eq = 1./np.sqrt(2)*np.sqrt((Sigma_stockMAX[:,2,instant]-Sigma_stockMAX[:,3,instant])**2 + (Sigma_stockMAX[:,3,instant])**2 + (-Sigma_stockMAX[:,2,instant])**2 + 6*Sigma_stockMAX[:,5,instant]**2)


    VM_MAX_eq_interpolate = interpolate(VM_MAX_eq, vtx, wts).reshape(X.shape[0],X.shape[0])

    plt.legend(loc='upper center', shadow=True)

    contour = plt.contour(XX_field[20:480,20:480], YY_field[20:480,20:480],ndimage.filters.gaussian_filter(VM_MAX_eq_interpolate[20:480,20:480], 5),colors=(cmap(instant/ np.float(Sigma_stockMAX.shape[2])),),levels = [320],linestyles=('-',),linewidths=(2,))

plt.savefig(name+ '_0' + test[instant][81:110] + ".png", dpi=250)

我试图在循环中添加这部分，但它不起作用：

labels = ['line1', 'line2','line3','line4',
           'line5', 'line6', 'line6', 'line6', 'line6', 'line6']

for i in range(len(labels)):
    contour.collections[instant].set_label(labels[instant])

Answer 1

我用过：http://matplotlib.org/examples/pylab_examples/contour_demo.html

cmap = plt.cm.hot
lines = []
labels = []
i = 0
instant = 0
for instant in range(0,Sigma_stockMAX.shape[2]):

    name = 'test'

    VM_MAX_eq = 1./np.sqrt(2)*np.sqrt((Sigma_stockMAX[:,2,instant]-Sigma_stockMAX[:,3,instant])**2 + (Sigma_stockMAX[:,3,instant])**2 + (-Sigma_stockMAX[:,2,instant])**2 + 6*Sigma_stockMAX[:,5,instant]**2)

    VM_MAX_eq_interpolate = interpolate(VM_MAX_eq, vtx, wts).reshape(X.shape[0],X.shape[0])

    contour = plt.contour(XX_field[20:480,20:480], YY_field[20:480,20:480],ndimage.filters.gaussian_filter(VM_MAX_eq_interpolate[20:480,20:480], 5),colors=(cmap(instant/ np.float(Sigma_stockMAX.shape[2])),),levels = [320],linestyles=('-',),linewidths=(2,))
    lines.extend(contour.collections)
    labels.extend(['line'+str(i+j) for j in range(len(contour.collections))])
    i += len(contour.collections)
plt.legend(lines, labels, loc='upper center', shadow=True)
plt.savefig(name+ '_0' + test[instant][81:110] + ".png", dpi=250)