1.5 - Numpy的方法中,axis参数的理解
1.5.1 Numpy常用方法
在Numpy常用方法如:sum,argmin等方法中,经常出现参数 axis。【axis表示维度,从0开始,含义是:如果设置axis=i,则计算会沿着 i 变化的方向操作。】
示例:
import sys import numpy as np a = np.arange(1, 7).reshape(2, 3) print(a) print(np.sum(a, axis=0).shape) # sum操作会降维, axis=0 对应的维度消失, shape为 (3,) print(np.sum(a, axis=1).shape) # sum操作会降维, axis=1 对应的维度消失, shape为 (2,) print('-------------------------------------------') b = np.arange(24).reshape(2, 3, 4) print(b) print(b.sum(axis=0)) # 结果的shape为 (3, 4) print(b.sum(axis=1)) # 结果的shape为 (2, 4)