Matplotlib xy线图图例

Question

如何在IPython笔记本内的Matplotlib中将图例添加到xy线图中？我目前的尝试：

x = np.linspace(0, 3*np.pi, 500)
a = plt.plot(x, np.sin(x**2))
b = plt.plot(x, np.sin(x**0.5))
plt.legend([a,b], ['square', 'square root'])

这样做，我得到以下错误：

/Users/mc/.virtualenvs/kaggle/lib/python2.7/site-packages/matplotlib/legend.py:613：用户警告：图例不支持[]使用代理艺术家代替。

http://matplotlib.sourceforge.net/users/legend_guide.html#using-代理艺术家

（str（orig_handle），）） /Users/mc/.virtualenvs/kaggle/lib/python2.7/site-packages/matplotlib/legend.py:613： UserWarning： Legend不支持[]使用代理艺术家代替。

http://matplotlib.sourceforge.net/users/legend_guide.html#using-代理艺术家

（原柄）

如果我做plt.scatter而不是plt.plot，但是我想要一个线图而不是x，y点，这个命令就可以工作。

Answer 1

这样做怎么样？

x = np.linspace(0, 3*np.pi, 500)
fig, ax = plt.subplots()
a = ax.plot(x, np.sin(x**2), label = 'square')
b = ax.plot(x, np.sin(x**0.5), label = 'square root')
handles, labels = ax.get_legend_handles_labels()
ax.legend(handles, labels)

为了得到这个：