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)

 

posted @ 2024-04-02 18:23  橘子葡萄火龙果  阅读(25)  评论(0编辑  收藏  举报