平时用的sum应该是默认的axis=0 就是普通的相加,当加入axis=1以后就是将一个矩阵的每一行向量相加。
例如:
1 >>>import numpy as np 3 >>>np.sum([[0,1,2],[2,1,3],axis=1) 5 array([3,6])
1 c = np.array([[0, 2, 1], [3, 5, 6], [0, 1, 1]]) 2 print c.sum() 3 print c.sum(axis=0) 4 print c.sum(axis=1) 5 结果分别是:19, [3 8 8], [ 3 14 2] 6 axis=0, 表示列。 7 axis=1, 表示行。
时间: 2024-11-03 20:47:18