sum(axis=1)是将一个矩阵的每一行向量相加
sum(axis=0)是将一个矩阵的每一列向量相加
对一维数组,只有第0轴,没有第1轴
例如:
import numpy as np
np.sum([[0,1,2],[2,1,3],axis=1)
结果就是:array([3,6])
c = np.array([[0, 2, 1], [3, 5, 6], [0, 1, 1]])
print c.sum()
print c.sum(axis=0)
print c.sum(axis=1)
结果分别是:19, [3 8 8], [ 3 14 2]
axis=0, 表示列。
axis=1, 表示行。