numpy.argmax()

weixin_33968104

于 2018-08-11 21:31:00 发布

阅读量68

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/tectal/p/9461178.html

本文详细介绍了 numpy 中的 argmax 函数用法，包括参数解释、返回值说明及多个示例。通过这些示例，读者可以了解到如何使用该函数获取数组中最大值的索引，适用于一维和多维数组的不同应用场景。

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

numpy.argmax(a, axis=None, out=None)
返回沿轴axis最大值的索引

Parameters:
a : array_like -数组
axis : int, 可选 -默认情况下，索引的是平铺的数组，否则沿指定的轴。
out : array, 可选 -如果提供，结果以合适的形状和类型被插入到此数组中。

Returns:
index_array : ndarray of ints -索引数组。它具有与a.shape相同的形状，其中axis被移除。

例子：

 1 >>> a = np.arange(6).reshape(2,3)
 2 >>> a
 3 array([[0, 1, 2],
 4        [3, 4, 5]])
 5 >>> np.argmax(a)
 6 5
 7 >>> np.argmax(a, axis=0)#0代表列
 8 array([1, 1, 1])
 9 >>> np.argmax(a, axis=1)#1代表行
10 array([2, 2])
11 >>>
12 >>> b = np.arange(6)
13 >>> b[1] = 5
14 >>> b
15 array([0, 5, 2, 3, 4, 5])
16 >>> np.argmax(b) # 只返回第一次出现的最大值的索引
17 1