擅长:python、mysql、java
<p>{a1}方法应该会有所帮助</p>
<p><strong>更新</strong></p>
<p>(阅读评论后)我相信<code>argmax()</code>方法也适用于多维数组。链接文档给出了一个示例:</p>
<pre><code>>>> a = array([[10,50,30],[60,20,40]])
>>> maxindex = a.argmax()
>>> maxindex
3
</code></pre>
<p><strong>更新2</strong></p>
<p>(感谢<a href="https://stackoverflow.com/users/224671/kennytm">KennyTM</a>的注释)您可以使用<code>unravel_index(a.argmax(), a.shape)</code>以元组的形式获取索引:</p>
<pre><code>>>> from numpy import unravel_index
>>> unravel_index(a.argmax(), a.shape)
(1, 0)
</code></pre>