import numpy
a = numpy.array([ [ [1,3,4],
[2,1,3],
[1,6,7] ],
[ [1,2,3],
[2,3,4],
[4,5,6] ] ])
b = a.sum()
c = a.sum(axis=0)
d = a.sum(axis=1)
e = a.sum(axis=2)
变量a为2*3*3维的高维数组
b为直接对所有的数字进行“暴力”相加。
>>> b
58
c为对高维数组a的第1维度(第一级方括号所包含的内容,内容为第二级方括号整体,发现二级括号有两个,直接将两个二级括号里面的内容对应相加)进行相加,加完之后为3*3的大小:
>>> c
array([[ 2, 5, 7],
[ 4, 4, 7],
[ 5, 11, 13]])
d为对高维数组a的第2维度(第二级方括号里面的内容,内容为第三级方括号整体,发现有三个三级括号,直接将三个三级括号对应相加)进行相加,加完之后为2*3大小:
>>> d
array([[ 4, 10, 14],
[ 7, 10, 13]])
e为对高维数组a的第3维度(第三级方括号里面的内容,里面为三个数值,直接相加即可)进行相加,加完之后为2*3大小:
>>> e
array([[ 8, 6, 14],
[ 6, 9, 15]])