numpy数组轴详解

概述

按照图一中aixs=0，对aixs=0上下对应的数据进行相加在学习numpy的时候，最难理解的就是轴的概念，我们知道坐标系中有轴的概念，那么两个轴是否有关联呢？为了便于理解，特写此博客进行梳理。

正文

首先数组的维数比较好理解，下面我们创建一个数组：

import numpy as np
# 创建一个三维数组
b=np.arange(24).reshape(4,3,2)

打印结果：

[[[ 0  1]
  [ 2  3]
  [ 4  5]]

 [[ 6  7]
  [ 8  9]
  [10 11]]

 [[12 13]
  [14 15]
  [16 17]]

b 是一个三维数组：

1. 第一维有三个元素
2. 第二维有三个元素
3. 第三维有四个元素

上面的数据也可以用下列方式展示（图一）

 

对于下面按照aixs=0进行sum：

print(b.sum(0))

按照图一中aixs=0，对aixs=0上下对应的数据进行相加，数据从（4，3，2）降维到（3，2）

[[0+ 6+12+18=36   1+ 7+13+19=40]
 [2+ 8+14+20=44   3+ 9+15+21=48]
 [4+10+16+22=52   5+11+17+23=56]]

对于下面按照aixs=1进行sum：

print(b.sum(1))

按照图一中aixs=1，按照比1小的轴对数据进行划分（即aixs=0），然后对划分的每一部分中数据中的aixs=1上下对应的数据进行相加，数据从（4，3，2）降维到（4，2）

[[ 0+ 2+ 4=6   1+ 3+ 5=9]
 [ 6+ 8+10=24  7+ 9+11=27]
 [12+14+16=42 13+15+17=45]
 [18+20+22=60 19+21+23+63]]

对于下面按照aixs=2进行sum：

print(b.sum(2))

按照图一中aixs=2，按照比2小的轴对数据进行划分（即aixs=0，aixs=1），然后对划分的每一部分中数据中的aixs=2上下对应的数据进行相加，数据从（4，3，2）降维到（4，3）

[[ 0+ 1=1   2+ 3=5   4+ 5=9]
 [ 6+ 7=13  8+ 9=17 10+11=21]
 [12+13=25 14+15=29 16+17=33]
 [18+19=37 20+21=41 22+23=45]]

总结：

1. aixs的范围是0到数组的维数（不包括维数）
2. 轴的划分是按照维数进行
3. 相加时按照轴进行对象相加，但是不能跨越比当前轴低的轴