import numpy as np a = np.array([[1, 5, 5, 2],[9, 6, 2, 8],[3, 7, 9, 1]]) print(np.argmax(a, axis=0))#輸出每列最大元素所在下標(biāo) print(np.argmax(a, axis=1))#輸出每行最大元素所在下標(biāo)

總結(jié)

以上是生活随笔為你收集整理的argmax用法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。