Matplotlib--legend

显示图例

import matplotlib.gridspec as gridspecfrom mpl_toolkits.mplot3d import Axes3Dimport numpy as npimport matplotlib.pyplot as pltfrom matplotlib.ticker import MultipleLocator, FormatStrFormatter# 生成X数据x_data = np.linspace(-2,2,50)# Y数据y1 = np.sin(10*x_data)*np.exp(-x_data) - 0.5y2 = np.square(x_data)l1, = plt.plot(x_data, y1, c='r', lw=2, ls = '-.')l2, = plt.plot(x_data, y2, c='g')plt.xlabel('X')plt.ylabel('Y')plt.legend(handles=[l1, l2], labels=['$sin(x)*exp(x)$', '$x^2$'])plt.savefig('d:/f.png')plt.show()