Python NumPy教程-numpy

此参数定义了要查找其最大值的源数组。

axis: int（可选）

此参数定义了索引所在的轴，默认为展平的数组。

out: array（可选）

此参数定义了结果将要插入的ndarray。这将是适合存储结果的相同类型和形状。

返回值

此参数定义了一个ndarray，其中包含数组的索引。其形状与x.shape相同，但去掉了沿轴的维度。

示例 1：

Import numpy as np  
x = np.arange(20).reshape(4,5) + 7  
x  
y=np.argmax(a)  
y  

输出：

array([[ 7,  8,  9, 10, 11],
        [12, 13, 14, 15, 16],
        [17, 18, 19, 20, 21],
        [22, 23, 24, 25, 26]])
19

在上面的代码中

• 我们使用别名np导入了numpy模块。
• 我们使用np.arange()函数创建了一个数组’x’，形状为四行五列。
• 我们在每个元素中都加了7。
• 我们声明了变量’y’并将np.argmax()函数的返回值赋值给它。
• 我们在函数中传递了数组’x’。
• 最后，我们尝试打印变量’y’。

输出显示了数组中最大元素的索引。