numpy 的axis=0,1具体含义

首先看一下这个例子

import numpy as np 
a=np.arange(8).reshape(2,2,2)
print(a.sum(axis=0))
print(a.sum(axis=1))
print(a.sum(axis=2))
'''
输出如下:
[[ 4  6]
 [ 8 10]]
[[ 2  4]
 [10 12]]
[[ 1  5]
 [ 9 13]]

'''

 

 

 

 总的来说,就是确定了哪个维度之后,另外的维度的数据要保持不变,而确定的哪个维度只需要(0,1...x)x=该维度的shape()的值-1,比如说a.shaoe=(2,2,2),那么第一维在(0,1),其他维度的值不变

posted on 2020-12-08 09:51  小小喽啰  阅读(165)  评论(0编辑  收藏  举报