直接上例子,簡單易懂!
import numpy
a = numpy.array([
? ? [[1,2,3],
? ? [1,2,3],
? ? [2,4,6]],
? ? [[5,10,15],
? ? [10,20,30],
? ? [20,40,60]]
])
b = a.sum(axis=1)
c = a.sum(axis=0)
print(b)
print(c)
輸出結(jié)果為:[[ 4 8 12]
[ 35? 70 105]]
[[ 6 12 18]
[11 22 33]
[22 44 66]]
認真觀察我們會發(fā)現(xiàn),輸出結(jié)果b中第一行 4=1+1+2,即a中第一個矩陣第一列相加;
輸出結(jié)果c中第一行 6=1+5,12=2+10,即a中第一個矩陣第一行相加。
代碼為本人親自敲寫,都可以運行,轉(zhuǎn)載的朋友請注明出處http://www.itdecent.cn/writer#/notebooks/35294278/notes/43145349,謝謝!