对具有相同第一列值的 2D numpy 数组元素进行分组

我有一个像这样的 2D numpy 数组

[[ 569  897]
 [ 570  898]
 [ 570  900]
 [ 571  901]
 [ 571  905]
 [ 572  906]]

我想要要分组的第一列中具有相等值的元素按照以下方式一起。

[[  569  897]
 [[ 570  898]
  [ 570  900]]
 [[ 571  901]
  [ 571  905]]
 [  572  906] ]

我应该怎样做呢？

您可以使用np.unique得到分离索引，然后使用np.split实际上分裂-

np.split(a, np.unique(a[:,0], return_index=1)[1][1:],axis=0)

或者，通过切片和使用np.flatnonzero -

np.split(a, np.flatnonzero(a[1:,0] != a[:-1,0])+1,axis=0)

样本运行 -

In [63]: a
Out[63]: 
array([[569, 897],
       [570, 898],
       [570, 900],
       [571, 901],
       [571, 905],
       [572, 906]])

In [64]: out = np.split(a, np.flatnonzero(a[1:,0] != a[:-1,0])+1,axis=0)

In [65]: out[0]
Out[65]: array([[569, 897]])

In [66]: out[1]
Out[66]: 
array([[570, 898],
       [570, 900]])

In [67]: out[2]
Out[67]: 
array([[571, 901],
       [571, 905]])

In [68]: out[3]
Out[68]: array([[572, 906]])