4

ループを使用せずに、numpy配列の軸に沿ってk-minimum値のインデックスを返す方法はありますか?

4

1 に答える 1

7
import numpy as np
x = np.array([[5, 2, 3],[1, 9, 2]]) # example data
k = 2 # return the indices of the 2 smallest values
np.argsort(x, axis=1)[:,0:k] # by row

array([[1, 2],
       [0, 2]])
于 2011-12-29T15:05:19.613 に答える