【Python之numpy库】6.axis轴的概念

如果解决了你的问题，点个赞再走嘛٩(๑❛ᴗ❛๑)۶

 二维数组中，有两个轴，axis=0表示列方向，axis=1表示行方向，放在函数参数中，我更倾向于：函数操作的方向

 举个例子：np.sum(x,axis,keepdims)求和函数

# 注：若没有keepdims= 那么一定会返回一维数组
# 记忆方法：若axis=1，则一定返回行数为1的二维行向量，则是列相加
# 与cvxpy库应该是一样的


import numpy as np

x = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])
b = np.sum(x, axis=1)  # 把每一行的3个列向量压缩，即对列向量求和，返回一个一维数组

d = np.sum(x, axis=0)

print('b = ', b)

print('d = ', d)

结果：

b =  [ 6 15 24]
d =  [12 15 18]

原因：

axis=1是表示行方向的操作，故沿着行方向将这个矩阵“压扁”求和（不知道能不能明白我这个比喻），所以

6=1+2+3

15=4+5+6

24=7+8+9

axis=0时同理

由于函数特性，因此返回的是一维向量