关于scatter画图分离数据问题

Question

问题遇到的现象和发生背景

ax1.scatter(x[:,0],x[:,1],cmap="tab20c",s=2,c=y,marker='.')

其中x是一个三维数组，y是一个一维数组
出来的一张图（如下）被颜色为了三个部分
我想知道怎么分别得到各个部分的数据，谢谢

Answer 1

• 如果你知道y里面有哪些确定的类别，比如y里面的值只有1，2，3；你可以通过x[y==1]索引y=1的类别，然后以此类推
• 如果你不知道有哪些类别的话，你可以使用numpy的unique(y)查看y里面有哪些值