【发布时间】：2012-01-29 22:53:59
【问题描述】：
有没有办法在不使用循环的情况下沿 numpy 数组的轴返回 k 最小值的索引？
【问题讨论】：
标签： python arrays numpy indices minim
【发布时间】：2012-01-29 22:53:59
【问题描述】：
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]])
【讨论】：