使用 pandas 和 matplotlib.pyplot 创建图例

Question

这是我第一次尝试使用 python 进行绘图，但在创建图例时遇到了问题。

这些是我的导入：

import matplotlib.pyplot as plt
import pandas

我这样加载我的数据：

data = pandas.read_csv( 'data/output/limits.dat', sep=r"\s+", encoding = 'utf-8' )

并像这样绘制它：

axdata = data.plot( label = '$|U|^{2}$' , x = 'mass', y = 'U2',
                    style = '-s', markeredgecolor = 'none' )

显然 axdata 现在是 AxesSubplot。

现在我想创建一个像 here 这样描述的图例：

plt.legend( (line1), ('label1') )

但我不知道如何从AxesSubplot 中提取line 对象

plt.legend() 自己的作品，但我只希望我的一些台词出现在图例中。这是正确的方法吗？我可以在这里使用其他命令吗？

编辑：

例如，如果我尝试：

plt.legend( [axdata], ['U2'])

我得到错误：

~/.virtualenvs/science/lib/python3.3/site-packages/matplotlib/legend.py:613:
UserWarning: Legend does not support Axes(0.125,0.1;0.775x0.8)
Use proxy artist instead.

http://matplotlib.sourceforge.net/users/legend_guide.html#using-proxy-artist

(str(orig_handle),))

我还没有弄清楚代理艺术家是什么，但我认为它是您使用非默认图形对象时的工具，我认为这里可能不是这种情况，因为我正在尝试制作一个正常的 matlibplot 图。 “非默认”和“正常”这两个词是我的 - 我还不确定它们的含义。

另一个编辑：（因为我误读了评论）

plt.legend() 本身不会向控制台输出任何内容，但生成的绘图现在具有从绘图数据自动生成的图例。

Answer 1

我认为你想要做的是能够为你的情节上的一部分线条显示一个图例。应该这样做：

df = pd.DataFrame(np.random.randn(400, 4), columns=['one', 'two', 'three', 'four'])
ax1 = df.cumsum().plot()
lines, labels = ax1.get_legend_handles_labels()
ax1.legend(lines[:2], labels[:2], loc='best')  # legend for first two lines only

给予