numpy中的sum函数如何区分axis=0和axis=1

函数如何区分axis=0和axis=1"/>

numpy中的sum函数如何区分axis=0和axis=1

        这里的主要内容参照 你是猪猪猪儿虫的 的博客写的括号最大法，上面是链接。

        以二维数组举例，

import numpy as np
a=np.array([[0,1,2],[3,4,5],[6,7,8]])
print(a)
#二维数组，axis=0时最大单位块为数组，对应下标元素相加，其实就是计算每一列所有值的和
#axis=1时最大单位块是数值，将每部分数值相加，其实就是计算每一行所有值的和
print(a.sum(axis=0))
print(a.sum(axis=1))

a的输出如下：
[[0 1 2][3 4 5][6 7 8]]axis=0的输出如下：
[ 9 12 15]
axis=0对应最外层的[]，最大单位块为[0,1,2]、[3,4,5]和[6,7,8]。
将对应下标元素相加，则[0+3+6,1+4+7,2+5+8]=[9,12,15]
对二维数组来说，其实就是计算每一列的和。axis=1的输出如下：
[ 3 12 21]
axis=1对应第二层的[],最大单位块是数值，则直接计算。
[0+1+2,3+4+5,6+7+8]=[3,12,21]
对二维数组来说，其实就是计算每一行的和。