1. 程式人生 > >python中的sum函式

python中的sum函式

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, 表示行。