numpy的max(), argmax(), min(), argmin()函数

目录

一、max()函数

二、argmax()函数

(1)二维数组

(2)三维数组

 min()、argmin()函数类似，这里不做赘述

---

一、max()函数

简单来说就是求array里的最大值；

返回值：最大值；

示例：

import numpy as np

x = np.arange(6).reshape(2, 3)
print(x)

y = np.max(x)
print(y)

输出：

二、argmax()函数

简单来说，就是求取最大值的索引值；

返回值：最大值的索引

难点：它有多个维度，用参数axis表示

(1)二维数组

axis = 0

示例：

arg_axis0 = np.argmax(a=x, axis=0)
print(arg_axis0)

 输出：

图解：

 axis = 1是类似的，只不过是以y轴为基准进行比较；

返回值是：[2, 2]

(2)三维数组

类似的，按照不同的方向去进行比较；

这里推荐一篇比较好的博客：

深入理解numpy中argmax的具体使用_python_脚本之家 (jb51.net)

 min()、argmin()函数类似，这里不做赘述