numpy 获取特定数值的索引_如何在NumPy数组中获得N个最大值的索引？

对于多维数组，您可以使用 axis 关键字来沿预期轴应用分区 .

# For a 2D array

indices = np.argpartition(arr, -N, axis=1)[:, -N:]

并 grab 物品：

x = arr.shape[0]

arr[np.repeat(np.arange(x), N), indices.ravel()].reshape(x, N)

但请注意，这不会返回排序结果 . 在这种情况下，您可以沿预期轴使用 np.argsort() ：

indices = np.argsort(arr, axis=1)[:, -N:]

# Result

x = arr.shape[0]

arr[np.repeat(np.arange(x), N), indices.ravel()].reshape(x, N)

这是一个例子：

In [42]: a = np.random.randint(0, 20, (10, 10))

In [44]: a

Out[44]:

array([[ 7, 11, 12, 0, 2, 3, 4, 10, 6, 10],

[16, 16, 4, 3, 18, 5, 10, 4, 14, 9],

[ 2, 9, 15, 12, 18, 3, 13, 11, 5, 10],

[14, 0, 9, 11, 1, 4, 9, 19, 18, 12],

[ 0, 10, 5, 15, 9, 18, 5, 2, 16, 19],

[14, 19, 3, 11, 13, 11, 13, 11, 1, 14],

[ 7, 15, 18, 6, 5, 13,