2014-03-26 53 views
1

是否有一個很好的快速方法來爲numpy數組執行以下選擇?如何在列更改numpy.array中的值時選擇第一行

>>> A=np.array([[1,2], [2,2], [3,5], [4,5]]) 
>>> A 
array([[1, 2], 
     [2, 2], 
     [3, 5], 
     [4, 5]]) 

我想在第二列更改值時選擇第一行。對於上述陣列,結果將爲:

array([[1, 2], 
     [3, 5]]) 

回答

2
>>> xs = np.array([[1,2], [2,2], [3,5], [4,5]]) 
>>> j = scipy.r_[True, xs[:-1,1] != xs[1:,1]] # or np.concatenate here 
>>> xs[j,:] 
array([[1, 2], 
     [3, 5]]) 
相關問題