Python学习笔记

您所在的位置:网站首页 axis命令是什么意思 Python学习笔记

Python学习笔记

2023-10-28 05:49| 来源: 网络整理| 查看: 265

在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