图例可以为可视化赋予实际含义,为不同的图标元素附上明确说明。我们前面看到了一些简单的图例创建例子;本小节中我们来介绍一下在 Matplotlib 中自定义图例的位置和进行美化的方法。
可以使用plt.legend()函数来创建最简单的图例,这个函数能自动创建任何带有标签属性的图表元素的图例:
import matplotlib.pyplot asplt
plt.style.use('classic')
import numpyasnp
x= np.linspace(0, 10, 1000)
fig, ax=plt.subplots()
ax.plot(x, np.sin(x),'-b', label='Sine')
ax.plot(x, np.cos(x),'--r', label='Cosine')
ax.axis('equal')
leg=ax.legend();
plt.show()
但除此之外还有很多能自定义图例的方法。例如,我们可以指定图例位置并且去除边框:
ax.legend(loc='upper left', frameon=False)
fig
我们可以使用ncol属性设置图例中每行的列数:
ax.legend(frameon=False, loc='lower center', ncol=2)
fi