numpy 轴的一些理解

来源:互联网 发布:python split 多个字符 编辑:程序博客网 时间:2024/05/16 17:01

numpy中的一些统计函数往往需要指定轴axis
如:
np.sum(a,axis = 1)
那么这个axis=1实际上是说,只对第一轴进行相加操作。比如说:

a = np.array([[1,2,3],[1,2,3]])print(np.sum(a,axis = 1))>>>[6 6]print(np.sum(a,axis = 0))>>>[2 4 6]print(np.sum(a))>>>12

可以看到,当axis=0时,即相加只对第一维度,第一维度的数为[1,2,3]和[1,2,3],所以这两个矩阵相加为[2,4,6]。
当axis=1时,即相加只对第二维度,第二维度的数相加,第一维不变,即为[1+2+3,1+2+3]=[6,6]。
当不加axis参数时,把所有元素相加,即为12。