当前位置 博文首页 > numpy的sum函数的axis和keepdim参数详解

    numpy的sum函数的axis和keepdim参数详解

    作者:油炸冰淇凌 时间:2021-05-24 18:24

    一.axis

    axis就是指定轴。

    三维数组可看作元素是二维数组的一维数组,二维数组可看作元素是一维数组的一维数组 。(这么理解就舒服了!)

    例:

    axis=2就是三维数组对最里面那一层即每个一维数组内部进行求和。

    axis=0就是对最外面那一层的元素之间作和。

    例子戳这里

    二.keepdim

    可以理解为'keepdims = True'参数是为了保持结果的维度与原始array相同,即keep dimension 保持维度.

    import numpy as np
     
    b=np.arange(12)
    b=b.reshape(2,6)
    print(b)
    print('b中的元素之和:',np.sum(b))
    #即在b的第一个轴上进行加和,相当于压缩行,也可以理解为二维矩阵的第一层括号里的东西加和
    #若axis=1则是压缩列,也就是对第二层括号里面的进行求和
    sum=np.sum(b,axis=0,keepdims=True)
    print(sum)

    运行结果:

    [[ 0  1  2  3  4  5]
     [ 6  7  8  9 10 11]]
    b中的元素之和: 66
    [[ 6  8 10 12 14 16]]
    最后一个输出指定axis=0,keepdim=True,可看到输出的是一个二维数组,如果不加keepdim=True,那么结果就是一维数组[ 6  8 10 12 14 16]

    js