Matplotlib 图例用颜色映射？答案

【问题标题】：Matplotlib 图例用颜色映射？
【发布时间】：2022-01-16 09:09:22
【问题描述】：

总的来说，我对 matplotlib 真的很困惑。我通常只使用 import matplotlib.pyplot as plt。

然后执行 plt.figure()、plt.scatter()、plt.xlabel()、plt.show() 等所有操作。但是后来我用谷歌搜索了如何做一些事情，用颜色映射图例，我得到了所有这些包括 ax 的例子。但是有 plt.legend() 并且 matplotlib 文档中的示例只显示 plt.legend(handles) 但没有向您显示应该是什么句柄。如果我想做斧头的事情，那么我必须重新编写所有代码，因为我想使用 plt，因为它更简单。

这是我的代码：

import matplotlib.pyplot as plt

colmap = {
    "domestic": "blue",
    "cheetah": "red",
    "leopard": "green",
    "tiger": "black"
}

colours = []

for i in y_train:
    colours.append(colmap[i])
    
plt.figure(figsize= [15,5])
plt.scatter(X_train[:,0], X_train[:,2],c=colours)
plt.xlabel('weight')
plt.ylabel('height')
plt.grid()
plt.show()

现在我想添加一个图例，它只显示与我的字典中相同的颜色。但如果我这样做：

plt.legend(["国产","猎豹","豹子","老虎"])

它仅在图例中显示“国内”，并且颜色为红色，这实际上与我对其进行颜色编码的方式不匹配。有没有办法做到这一点而不用“斧头”的东西重写一切？如果没有，我该如何适应斧头？我只写 ax = plt.scatter(....) 吗？

【问题讨论】：

请提供一些示例数据和您想要的输出。（见How to make good reproducible pandas examples）。

标签： python matplotlib legend

【解决方案1】：

没有提供数据，但是这段代码可以帮助你理解如何在 matplotlib 中为散点图添加颜色：

将 matplotlib.pyplot 导入为 plt 将 numpy 导入为 np

# data for scatter plots
x = list(range(0,30))
y = [i**2 for i in x]

# data for mapping class to color
y_train = ['domestic','cheetah', 'cheetah', 'tiger', 'domestic',
           'leopard', 'tiger', 'domestic', 'cheetah', 'domestic',
           'leopard', 'leopard', 'domestic', 'domestic', 'domestic',
           'domestic', 'cheetah', 'tiger', 'cheetah', 'cheetah',
           'domestic', 'domestic', 'domestic', 'cheetah', 'leopard',
           'cheetah', 'domestic', 'cheetah', 'tiger', 'domestic']

# color mapper
colmap = {
    "domestic": "blue",
    "cheetah": "red",
    "leopard": "green",
    "tiger": "black"
}

# create color array
colors = [colmap[i] for i in y_train]

# plot scatter    
plt.figure(figsize=(15,5))
plt.scatter(x, y, c=colors)
plt.xlabel('weight')
plt.ylabel('height')
plt.grid()
plt.show()

输出：

【讨论】：