plt.plot(w2,train2[0,:],label=‘A‘)

plt.plot(w2,train2[1,:],label=‘B‘)

plt.plot(w2,train2[2,:],label=‘C‘)

plt.legend()

plt.show()

label表示你想用的标记名称。

plt.legend()用于显示。