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

参数axis=0,1,2…