【Python】笨辦法理解高維數組求和 .sum(axis=0/1/2/...)

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]])