如何找到一个numpy数组的第n个最小值

时间:2019-02-20 17:53:03

标签: python arrays numpy

我想在二维numpy数组的最后一行中按元素的升序查找索引。一个例子如下。

ti = np.array([[255, 2, 255, 9], [4, 255, 6, 255], [23, 11, 5, 18]])

# output = function(ti[-1])
# output : 2, 1, 3, 0

我该如何实现? np.argmin仅给出一个最小值,而我找不到其他方法,包括使用分区(它将返回已排序的元素)。请帮助。

1 个答案:

答案 0 :(得分:-1)

我尝试了np.argsort,它给出了这种输出。如果有更多需要补充的见解,请其他人给予启发。

np.argsort(ti[-1])

输出:

array([2, 1, 3, 0], dtype=int64)