python - 使用索引分割 Numpy 数组

Question

我有 3D 数组

pcar=[[xa ya za]
      [xb yb zb]
      .
      .
      [xn yn zn]]

索引数组为

[0,1,0....,2]

给出了pcar中的行应该位于哪个簇中，以便我可以用不同的颜色绘制它。簇的最大值为 3。

输出应该是

clusters[0] = [[xa ya za], [xc yc zc], ...]
clusters[1] = [[xb yb zb], ...]
clusters[2] = [..., [xn yn zn]]

我想打印图中的点，每个簇点具有不同的颜色

Answer 1

据我了解，您想要生成如下所示的集群列表:

clusters[0] = [[xa ya za], [xc yc zc], ...]
clusters[1] = [[xb yb zb], ...]
clusters[2] = [..., [xn yn zn]]

使用 NumPy 索引很容易做到这一点。让索引数组称为indices。那么，

indices = np.asarray(indices)
num_clusters = 3
clusters = [pcar[indices==i] for i in xrange(num_clusters)]

会给你想要的结果。