sum就是求和,axis就是表示沿着axis方向求和。
在二维数组中:
axis=1按行的方向相加,返回每个行的值;
axis=0按列的方向相加,返回每个列的值。
import numpy as np
a = np.arange(8).reshape(2, 2, 2)
a
array([[[0, 1],
[2, 3]],
[[4, 5],
[6, 7]]])
# 按列相加
a.sum(axis=0)
array([2, 4])
# 按行相加
a.sum(axis=1)
array([1, 5])