如何使用字典键编写图例？答案

【问题标题】：How to write a legend using dictionary keys?如何使用字典键编写图例？
【发布时间】：2016-04-22 19:13:47
【问题描述】：

我正在尝试可视化以下数据：dictMy = {'apple' : [[0, 1], [0, 2], [3, 2]], 'pear' : [[2, 3], [3, 5], [0, 2]], 'peach' : [[2, 5], [3, 8], [0, 0]]} 这是我的代码：

from matplotlib.pyplot import cm
import matplotlib.pyplot as plt
import numpy as np


dictMy = {'apple' : [[0, 1], [0, 2], [3, 2]], 'pear' : [[2, 3], [3, 5], [0, 2]], 'peach' : [[2, 5], [3, 8], [0, 0]]}

color=iter(cm.rainbow(np.linspace(0,1,len(dictMy))))

for key in dictMy:
    curLabel = key
    c=next(color)
    for item in dictMy[key]:
        x = item[0]
        y = item[1]
        plt.scatter(x,y, c = c)
    plt.legend(str(curLabel))
plt.show()

这是我的输出：

所以，我真的不明白，为什么它以这种方式显示图例以及如何解决这个问题。我或多或少明白为什么它是所有键中的最后一个，但我不明白为什么它被分成字母。请帮忙。

【问题讨论】：

显然legend 函数需要一个可迭代的字符串，并且您提供一个字符串。当你遍历一个字符串时，你会得到单个字符。您可能不应该在循环内调用legend()。
@LevLevitsky 谢谢你的解释）

标签： python python-2.7 matplotlib

【解决方案1】：

目前，您正在为字典中的每个 key 调用 legend 函数。您不需要这样做 - 只需使用字典键标记散点图上的每种类型的点，然后调用 legend 函数。以下将起作用：

from matplotlib.pyplot import cm
import matplotlib.pyplot as plt
import numpy as np


dictMy = {'apple' : [[0, 1], [0, 2], [3, 2]],
          'pear' : [[2, 3], [3, 5], [0, 2]],
          'peach' : [[2, 5], [3, 8], [0, 0]],
         }
color=iter(cm.rainbow(np.linspace(0,1,len(dictMy))))

for key, c in zip(dictMy, color):
    for idx, item in enumerate(dictMy[key]):
        x = item[0]
        y = item[1]
        if idx == 0:
            plt.scatter(x, y, c=c, label=key)
        else:
            plt.scatter(x, y, c=c)

plt.legend()
plt.show()

输出

【讨论】：