簡單來說,對於一個m*n的矩陣,如果sum()的參數中沒有指定axis,就是將所有的數值加到一起,得到1*1的標量。如果是axis=0,就是按照列進行加,得到一個1*n的行向量;如果axis=1,表示矩陣每一個行相加,得到一個m*1的列向量。
例如:
import numpy as np
np.sum([[0,1,2],[2,1,3],axis=1)
結果就是:array([3,6])
a = np.array([[0, 2, 1]])
print a.sum()
print a.sum(axis=0)
print a.sum(axis=1)
結果分別是:3, [0 2 1], [3]
b = np.array([0, 2, 1])
print b.sum()
print b.sum(axis=0)
print b.sum(axis=1)
結果分別是:3, 3, 運行錯誤:'axis' entry is out of bounds
可知:對一維數組,只有第0軸,沒有第1軸