我有一个numpy.array
称为f的数据,我知道其中的最大值是,f_max=max(f)
但我想知道数组中与最大值相对应的索引。
我试过了:
count = 0
while (f[count]!=fmax)
conto ++
但我收到一个错误:
SyntaxError: invalid syntax
有人可以帮我吗?
最简单的方法是找到max
,然后寻找其索引。
>>> a = [1, 5, 2, 3, 4]
>>> val = max(a)
>>> a.index(val)
1
您还可以enumerate
用来获取索引列表以及与之对应的值,并从中选择最大值。
>>> list(enumerate(a))
[(0, 1), (1, 5), (2, 2), (3, 3), (4, 4)]
>>> index, _ = max(enumerate(a), key = lambda x: x[1])
>>> index
1
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句