如何考虑标签的颜色绘制不同颜色的配对数据

Question

我有两个列表，第一个是成对数据的列表（数据包含76对），第二个是对应的标签（标签包含0到11之间的76个数字）。我需要11种不同的颜色。 考虑到其标签，我需要使用特定的颜色绘制每个配对数据。 这两个列表如下：

data=[[2,6],[1,7],[3,8],[4,6],[7,9]] 
labels=[1,0,0,2,2]

我是python的初学者。任何帮助将不胜感激

Answer 1

import matplotlib.pyplot as plt

data=[[2,6],[1,7],[3,8],[4,6],[7,9]] 
labels=[1,0,0,2,2] #labels should have the same length as data
x = [i[0] for i in data] #separates the pairs into just the x component
y = [i[1] for i in data] #separates the pairs into just the y component

plt.scatter(x, y, c = labels, s = 50, cmap = 'rainbow')
plt.show()

“ c =标签”表示“使用标签值选择颜色” “ s = 50”确定点的大小，必要时进行调整，以便可以看到颜色 “ cmap ='rainbow'”确定您用于颜色的颜色图。

以下是带有matplotlib名称的颜色图：matplotlib colormaps