切片数组的numpy argmin（）返回原始索引？

Question

我有一个numpy的阵列

print(a) 
> array([ 0.  , 341.59065809, 215.33694527, 586.00085324]) 

，当我分析它，并尝试采取argmin（）

print(a[1,3]) 
> array([ 341.59065809, 586.00085324]) 
print(a[1,3].argmin()) 
> 0 

但是我想要得到的原始数组的索引（即1） 。我怎样才能做到这一点？

Answer 1

可以使用argmin输出索引到你原来的数组索引（即[1,3]）

a = array([ 0.  , 341.59065809, 215.33694527, 586.00085324]) 
    elems = [1,3] 
    min_ind = elems[a[elems].argmin()] #1