组2D numpy数组元素具有相等的第一列值

Question

我有一个像这样的二维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] ]

我应该怎么做？

Answer 1

您可以使用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]])