我想在二维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仅给出一个最小值,而我找不到其他方法,包括使用分区(它将返回已排序的元素)。请帮助。
答案 0 :(得分:-1)
我尝试了np.argsort,它给出了这种输出。如果有更多需要补充的见解,请其他人给予启发。
np.argsort(ti[-1])
输出:
array([2, 1, 3, 0], dtype=int64)