[英]numpy argmin() of sliced array return original index?
我有一个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)。 我该如何实现?
您可以使用argmin
输出来索引原始索引数组(即[1,3]
)
a = array([ 0. , 341.59065809, 215.33694527, 586.00085324])
elems = [1,3]
min_ind = elems[a[elems].argmin()] #1
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.