numpy中argmax的用法

原创于 2019-05-14 10:23:26 发布 · 1.7w 阅读

CC 4.0 BY-SA版权

3 篇文章

订阅专栏

本文详细解析了numpy库中argmax函数的使用方法，包括如何通过设置axis参数获取矩阵中每一行或每一列最大值的索引。通过具体代码示例，展示了argmax在不同维度上的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

axis=1

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