np.argmax()的使用
2.一维向量会根据索引(0开始),得到最大值的索引下标import numpy as npa=np.array([1,2,3,4,5])print(np.argmax(a)) # 4 2.二维向量axis=0 按列开始找最大值,下述例子中,max{1,6,11}=11 max{2,7,12}=12... max{5,10,15}=11所以按照列开始后 输出的是[2 2 2 2 2](索引0开始)axis=1 按行开始找最大值,同理argmax(a):寻找最大值的索引下标
原创
2022-04-05 21:02:53 ·
1511 阅读 ·
0 评论