import numpy
a = numpy.array([
[[5,10,15],
[10,20,30],
[20,40,60]],
[[5,10,15],
[10,20,30],
[20,40,60]]
])
#按行相加,此时第一、二行都是[5,10,15],[10,20,30],[20,40,60]
b = a.sum(axis=1)
#按列相加,此时第一列是[5,10,15],第二列是[10,20,30],第三列是[20,40,60]
c = a.sum(axis=0)
print(b)
print(c)
结果:
[[ 35 70 105]
[ 35 70 105]]
[[ 10 20 30]
[ 20 40 60]
[ 40 80 120]]
总结:简单的理解,axis=1按行的方向相加,返回每个行的值;axis=0按列相加,返回每个列的值。