Question

使用matplotlib制作图例有几种方法。可能是更简单的方式：

>>> line_up, = plt.plot([1,2,3], label='Up')
>>> line_down, = plt.plot([3,2,1], label='Down')
>>> plt.legend()
<matplotlib.legend.Legend object at 0x7f527f10ca58>
>>> plt.show()

另一种方式可能是：

>>> line_up, = plt.plot([1,2,3])
>>> line_down, = plt.plot([3,2,1])
>>> plt.legend((line_up, line_down), ('Up', 'Down'))
<matplotlib.legend.Legend object at 0x7f527eea92e8>
>>> plt.show()

这最后一种方式似乎只适用于支持迭代的对象：

>>> line_up, = plt.plot([1,2,3])
>>> plt.legend((line_up), ('Up'))
/usr/lib64/python3.4/site-packages/matplotlib/cbook.py:137: MatplotlibDeprecationWarning: The "loc" positional argument to legend is deprecated. Please use the "loc" keyword instead.
  warnings.warn(message, mplDeprecation, stacklevel=1)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib64/python3.4/site-packages/matplotlib/pyplot.py", line 3519, in legend
    ret = gca().legend(*args, **kwargs)
  File "/usr/lib64/python3.4/site-packages/matplotlib/axes/_axes.py", line 496, in legend
    in zip(self._get_legend_handles(handlers), labels)]
TypeError: zip argument #2 must support iteration

如果我想绝对使用只有一条曲线的第二种方式......我可以做什么？

Answer 1

我认为这样做的原因是定义一个项目元组，您将使用语法(line_up,)。请注意尾随的逗号。

import matplotlib.pyplot as plt
line_up, = plt.plot([1,2,3])
plt.legend((line_up,), ('Up',))
plt.show()

如果您不想包含尾随逗号，也可以使用列表。例如：

import matplotlib.pyplot as plt
line_up, = plt.plot([1,2,3], label='my graph')
plt.legend(handles=[line_up])
plt.show()

TypeError：zip参数＃2必须支持迭代，使用matplotlib.pyplot.legend（）

1 个答案: