Python学习笔记——参数axis=0,1,2...

在Python程序中，当需要对一些多维的数组或列表进行操作时，经常碰到操作函数中有axis这个参数。那么这个参数应该如何理解呢？axis=0/1/2该怎么区别？下面通过一段示例程序来学习。

（1）首先定义一个2x3x4的三维数组a：

a = [[[1,2 ,3 ,4 ],
      [5,6 ,7 ,8 ],
      [9,10,11,12]],
     [[13,14,15,16],
      [17,18,19,20],
      [21,22,23,24]]]

打印a，输出如下：

[[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]], [[13, 14, 15, 16], [17, 18, 19, 20], [21, 22, 23, 24]]]

（2）用numpy中的sum函数对a求和，先输入axis=0：

sum_0 = np.sum(a,axis = 0)
print(sum_0)

输出结果是一个3x4的数组，如下：

[[14 16 18 20]
 [22 24 26 28]
 [30 32 34 36]] 

（3）再用sum函数对a求和，输入axis=1：

sum_1 = np.sum(a,axis = 1)
print(sum_1)

输出结果是一个2x4的数组，如下：

[[15 18 21 24]
 [51 54 57 60]] 

（4）最后用sum函数对a求和，输入axis=2：

sum_2 = np.sum(a,axis = 2)
print(sum_2)

输出结果是一个2x3的数组，如下：

[[10 26 42]
 [58 74 90]] 

总结一下，对于一个shape为i*j*k的三维数组a，当执行函数function(a, axis=0)时，相当于在第1个dimension上进行function运算，其他维度的形状保持不变，运算结果的shape为原数组a去掉第一个dimension的sahpe（即j*k），当axis=1或2时同理。当然，axis的取值不限于0/1/2，还可以拓展到高维数组。