【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]])
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值