numpy中axis参数理解

前言

在学习axis时,其为0或者为1的时候,到底是行还是列。初学的时候感觉好像numpy和python中不一样,0有的时候代表行,有的时候代表列,经常犯错。其实是自己对其理解有误。下面开始来讲解。

numpy当中axis的值表示的是这个多维数组维度的下标,比如有一个二维数组a,a的shape是(5,6),也就是说a有5行6列,axis=0表示的就是[5,6]中的第一维,也就是行,axis=1表示的是[5,6]中的第二个维度,也就是列。

简单的来记就是axis=0代表往跨行(down),而axis=1代表跨列(across)。

axis=0是对行进行操作,axis=1是对列进行操作。

# 创建一个矩阵a,a的维度为2行3列
>>> a = np.array([[1,2,3],[7,8,9]])
>>> print a
[[1 2 3]
 [7 8 9]]


# b表示沿着axis=0(行)这条轴取max,对行进行操作,跨行操作,得到的结果就是把输入数组的'行'给消除了,2行变1行,若是求和,则就是[8 10 12]
>>> b = a.max(axis=0)
>>> print b
[7 8 9]


# c表示沿着axis=1(列)这条轴取max,得到的结果就是把输入数组的'列'给消除了,3列变1列,若是求和,则就是[6 24]
>>> c = a.max(axis=1)
>>> print c
[3 9]

扩展为多维;若有三个维度,则:axis=0代表对维度为0进行操作,axis=1代表对维度为1进行操作,axis=2代表对维度为2进行操作

猜你喜欢

转载自blog.csdn.net/comway_Li/article/details/81158902