有没有一种很好的快速方法来为numpy数组进行以下选择?
>>> 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]])
>>> 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]])