Numpy按轴计算

a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])

如果二维数组a展示到图上就是：

因为a[0]为[1,2,3,4]，a[1]为[5,6,7,8],a[2]为[9,10,11,12]，所以[1,2,3,4]在0轴0的位置，[5,6,7,8]在0轴1的位置，而[9,10,11,12]在0轴2的位置。

1是a[0][0]，2是a[0][1]， 3是a[0][2]，4是a[0][3]，所以[1,2,3,4]按照自己的下标顺序沿着1轴依次展开。5,6,7,8和9,10,11,12也是同样道理，按照自己的下标顺序沿着1轴依次展开。

数组计算时，声明axis=0，就时沿着0轴方向进行；axis=1，就是沿着1轴方向进行。

例如：a.sum(axis=0)

计算就是沿着0轴方向进行元素求和，也就是计算每列的和：

最终得到的数组是[15,18,21,24]

 

再比如 a.mean(axis=1) 

计算就是沿着1轴方向进行元素求平均值，也就是计算每行的平均值：

最终得到的数组是[2.5, 6.5, 10.5]