400-1100-266
9:00-24:00（工作日）
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]
用法：求的是最大值对应的索引
案例如下
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
标签：红河 上海 乐山 沈阳 河南 长治 沧州 新疆
上一篇：Python3 类型标注支持操作
下一篇：python str()如何将参数转换为字符串类型
