numpy中sum函數求和時參數axis=0和axis=1的含義

簡單來說,對於一個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軸

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章