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进行操作