直接上例子,简单易懂!
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)
输出结果为:[[ 4 8 12]
[ 35 70 105]]
[[ 6 12 18]
[11 22 33]
[22 44 66]]
认真观察我们会发现,输出结果b中第一行 4=1+1+2,即a中第一个矩阵第一列相加;
输出结果c中第一行 6=1+5,12=2+10,即a中第一个矩阵第一行相加。
代码为本人亲自敲写,都可以运行,转载的朋友请注明出处https://www.jianshu.com/writer#/notebooks/35294278/notes/43145349,谢谢!