我有一个形状为(2000, 1)的Numpy向量，对于每行的每个元素（x，y），我在一个形状为1,2,3,4,5,6的Numpy向量中有一个关联的标签。标签只是一个整数，可以是以下prob.Constraints.cons1 = ((x(1,1)-x0)/dt <= vmax);
我怎样才能得到一个散点图，其中每个点的颜色根据其标签而不同？
答案 0 :(得分：-1)
您可以直接在分散函数的color参数中提供整数标签
import numpy as np
import matplotlib.pyplot as plt
plt.scatter( data[:,0], data[:,1], c=labels )
plt.show()