python - 基于标签的颜色散点图

Question

我有一个 shape 的 numpy 向量(2000, 2)，对于每一行的每个元素 (x,y)，我在 shape 的 numpy 向量中有一个关联的标签(2000, 1)。标签只是一个整数，可以是以下任何一种1,2,3,4,5,6

我怎样才能有一个散点图，其中每个点的颜色根据其标签而不同？

标签： pythonmatplotlib

Answer 1

您可以通过直接在 scatter 函数的颜色参数中提供整数标签

import numpy as np
import matplotlib.pyplot as plt

plt.scatter( data[:,0], data[:,1], c=labels )
plt.show()