import numpy as npa = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])print(np.argmax(a))print(np.argmax(a, axis=0)) #豎著比較，返回行號(hào)print(np.argmax(a, axis=1)) #橫著比較，返回列號(hào)

輸出：

7[0 1 0 1][2 3]

用法：求的是最大值對(duì)應(yīng)的索引

案例如下

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持好吧啦網(wǎng)。