在numpy数组中找到最大值及其索引

Question

我有一个数组，如：

[ [0.0], [0.0020777732133865356], [0.0013433878775686026], [0.00021494206157512963], [8.955918747233227e-05], [0.0], [0.0], [1.7911837858264334e-05], [0.0], [1.7911837858264334e-05], [0.0], [0.0], [0.0], [0.0007702090078964829], [0.02625875361263752], [0.13904960453510284], [0.30124127864837646], [0.30514606833457947], [0.4224506914615631], [0.45712801814079285], [0.5807734131813049], [0.5874545574188232], [0.695248007774353], [0.18126779794692993], [0.11689265072345734], [0.07207723706960678], [0.06512743979692459], [0.06016586348414421], [0.04363323748111725], [0.030235182493925095], [0.03095165640115738], [0.028963441029191017], [0.03578785061836243], [0.029267942532896996]]

我想找到最大值及其索引。我已经搜索过这个，但没有一个符合我的问题。有什么解决办法吗？谢谢。

Answer 1

怎么了

In [6]: ind = np.argmax(a)

In [7]: a[ind]
Out[7]: array([ 0.69524801])

由于您有一个二维数组，您可能更喜欢：

In [9]: a[ind][0]
Out[9]: 0.69524800777435303