菜鸟进阶:(6)numpy的argmax用法

最新推荐文章于 2023-10-24 10:14:50 发布

VIP文章会编程的漂亮小姐姐

最新推荐文章于 2023-10-24 10:14:50 发布

阅读量181

点赞数

分类专栏： AI

本文链接：https://blog.csdn.net/u014229742/article/details/108645037

版权

argmax返回的是最大数的索引.argmax有一个参数axis,默认是0,表示每一列的最大值的索引 axis=1表示每一行的最大值的索引

`axis` = 0 列

import numpy as 
npa = np.array([[1, 5, 5, 2],            
                [9, 6, 2, 8],            
                [3, 7, 9, 1]])
print(np.argmax(a, axis=0))

结果：

最低0.47元/天解锁文章

优惠劵

会编程的漂亮小姐姐

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
菜鸟进阶:(6)numpy的argmax用法

解释还是从一维数组出发.看下面的例子.import numpy as npa = np.array([3, 1, 2, 4, 6, 1])print(np.argmax(a))4argmax返回的是最大数的索引.argmax有一个参数axis,默认是0,表示第几维的最大值.看二维的情况.import numpy as npa = np.array([[1, 5, 5, 2],[9, 6, 2, 8],[3, 7, 9, 1]])print(np.argmax(a, axis=0))[
复制链接

扫一扫