小编abl*_*rks的帖子

我想找到一个不是无穷大的1D numpy数组中最大值的索引.我已经尝试过argmax,但是当我的数组中存在无穷大值时,它只返回该索引.我提出的代码看起来非常h​​acky和不安全.有更好的解决方案吗？

import numpy as np
Y=np.array([2.3,3.5,np.inf,4.4,np.inf,2.5])

idx=np.where(Y==np.max(Y[np.isfinite(Y)]))[0][0]