np.max(等价于np.amax)用法

np.max与np.amax是同名函数

https://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.max.html

np.amax

numpy.amax(a, axis=None, out=None, keepdims=<no value>, initial=<no value>)[source]

返回数组a的最大值。
常用参数axis，即沿着指定轴
例

import numpy as np


a=np.array([[0, 1, 3, 4, 2],
            [5, 6, 7, 1, 0]])

print(np.amax(a, axis=0))
print(np.amax(a, axis=1))

[5 6 7 4 2]
[4 7]