切片数组的numpy argmin()返回原始索引?

时间:2016-03-02 01:51:56

标签: python arrays numpy

我有一个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)。我怎样才能做到这一点?

1 个答案:

答案 0 :(得分:1)

您可以使用argmin输出索引到原始索引数组(即[1,3]

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