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

Question

我有一个纽比阵列

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

当我把它切成薄片，试着用argmin（）

^{pr2}$

但是我想得到原始数组的索引（即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