a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])print(np.argmax(a))print(np.argmax(a, axis=0)) #竖着比较,返回行号print(np.argmax(a, axis=1)) #横着比较,返回列号 AI代码助手复制代码 输出: 7 [0 1 0 1] [2 3] Python np.argmax()的用法 用法:求的是最大值对应的索引 案...
import numpy as np a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]]) print(np.argmax(a)) print(np.argmax(a, axis=0)) #竖着比较,返回行号 print(np.argmax(a, axis=1)) #横着比较,返回列号 输出: 7 [0 1 0 1] [2 3] Python np.argmax()的用法 用法:求的是最大值对应的...