Python学习笔记 |
您所在的位置:网站首页 › axis命令是什么意思 › Python学习笔记 |
在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,还可以拓展到高维数组。 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |