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），其他维度的值不变