python学习笔记(一)argmax()

1,argmax()

(1)针对一维数组:返回数组中最大值元素的索引位置,如下代码所示,输出结果为元素6的索引值:4

import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
print(np.argmax(a)) #输出结果为4

(2)针对二维数组:参数axis可取(0,1),默认是0,表示数组第几维的最大值。下面分别针对axis=0,axis=1进行举例解释

  • axis = 0,简单理解,返回结果为:数组中每一列最大值所在“行”索引值。如下代码所示:第一列最大值是9(所在“行”索引值是1),所以针对该列的输出值为1;第二列最大值为7(所在“行”索引值是2),所以针对该列的输出值为2;同理,针对第三列的输出值为2,针对第四列的输出值为1;所以最终的输出值为(1,2,2,1)
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)) #输出结果为(1,2,2,1)
  • axis = 1,简单理解,返回结果为:数组中每一行最大值所在“列”索引值。如下代码所示:第一行最大值为5(所在列索引值为1),所以针对第一行的输出值为1;同理,针对第二,第三行的输出值分别为:0,2
import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=1)) #输出结果为(1,0,2)

(3)针对三维的数组,推荐一篇博文  http://www.cnblogs.com/zhouyang209117/p/6512302.html
 

 

 

  • 8
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值