，其具有相等的第一列組2D numpy的數組元素的值

Question

我有一個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] ] 

我應該怎麼做？

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]])