python常用知识点小结(持续更新)

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


猜你喜欢

转载自blog.csdn.net/manduner/article/details/79493689