统计运算

使用之前使用的 temp

temp.max(axis=0)
array([1.1, 1.1, 1.1, 1.1])

也可以这样

np.max(temp, axis=-1)
array([1.1, 1.1, 1.1, 1.1])

还可以这样

np.argmax(temp, axis=-1)
array([0, 2, 1, 3])

Last updated