numpy数组某轴上k个最小值的索引

Question

有没有办法在一个numpy数组的某个轴上返回k个最小值的索引，而不使用循环呢？

Answer 1

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

当然可以！请把你想要翻译的内容发给我，我会帮你用简单易懂的语言解释清楚。