如何一次分配多个图例标签？

我有以下数据集：

x = [0, 1, 2, 3, 4]
y = [ [0, 1, 2, 3, 4],
      [5, 6, 7, 8, 9],
      [9, 8, 7, 6, 5] ]

现在我用以下方法绘制它：

import matplotlib.pyplot as plt
plt.plot(x, y)

但是，我想用此命令标记 3 个 y 数据集，这会在以下情况下引发错误.legend()叫做：

lineObjects = plt.plot(x, y, label=['foo', 'bar', 'baz'])
plt.legend()

File "./plot_nmos.py", line 33, in <module>
  plt.legend()
...
AttributeError: 'list' object has no attribute 'startswith'

当我检查lineObjects:

>>> lineObjects[0].get_label()
['foo', 'bar', 'baz']
>>> lineObjects[1].get_label()
['foo', 'bar', 'baz']
>>> lineObjects[2].get_label()
['foo', 'bar', 'baz']

Question

有没有一种优雅的方法来分配多个标签，只需使用.plot() method?

您可以迭代线条对象列表，以便单独分配标签。使用内置 python 的示例iter功能：

lineObjects = plt.plot(x, y)
plt.legend(iter(lineObjects), ('foo', 'bar', 'baz'))`

Edit:更新到 matplotlib 1.1.1 后，它看起来像plt.plot(x, y)，其中 y 作为列表的列表（由问题作者提供）不再起作用。在将 y 传递为numpy.array（假设（numpy）[http://numpy.scipy.org/] 之前已导入）。

在这种情况下，使用plt.plot(x, y)（如果 2D y 数组中的数据排列为列 [轴 1]）或plt.plot(x, y.transpose())（如果 2D y 数组中的数据排列为行 [axis 0]）

Edit 2:正如@pelson 所指出的（见下面的评论），iter功能是不必要的并且简单plt.legend(lineObjects, ('foo', 'bar', 'baz'))完美运作