numpy sum的 axis的使用

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/TH_NUM/article/details/82027394

import numpy  as np

b=np.array([[[1,2,3],[4,5,6],[7,8,9]],[[1,2,3],[4,5,6],[7,8,9]]])
b=np.sum(b,axis=(0,2),keepdims=False)
print(b)
#如何理解？
#b的维度 [2,3,3]
#axis=(0,2)
#所以最终会变成[1,3,1]
#计算的过程：sum([:,k,:])只要循环第1维和第3维，最终的k是序列的index

a=np.array([[1,2,3],[4,5,6]])
a=np.sum(a,axis=(0))
print(a)

输出：

[12 30 48]
[5 7 9]

如何理解？

b的维度 [2,3,3]

np.sum(b,axis=(0,2),keepdims=False)

所以最终会变成[1,3,1]

计算的过程：sum([:,k,:])只要循环第1维和第3维，最终的k是序列的index