学堂 学堂 学堂公众号手机端

numpy库中的sum函数用于计算数组中元素的和。 用法

lewis 1年前 (2024-03-26) 阅读数 3 #技术

numpy库中的sum函数用于计算数组中元素的和。

用法: numpy.sum(a,axis=None,dtype=None,out=None,keepdims=)

参数说明:


  • a:数组
  • axis:指定计算的轴向,默认为None,即将所有元素相加。如果指定axis=0,则按列相加;如果指定axis=1,则按行相加。
  • dtype:指定返回的数据类型,默认为None,即保持原数组的数据类型。
  • out:指定输出的数组,用于保存计算结果。
  • keepdims:保持输出数组的维度。如果该参数为True,则结果具有与输入数组相同的维度,如果为False,则结果是一个简化的数组。

返回值: 返回一个数组或者标量,表示数组中元素的和。

示例:

importnumpyasnp a=np.array([1,2,3,4,5]) print(np.sum(a))#输出15 b=np.array([[1,2,3],[4,5,6]]) print(np.sum(b))#输出21 c=np.array([[1,2,3],[4,5,6]]) print(np.sum(c,axis=0))#输出[579] d=np.array([[1,2,3],[4,5,6]]) print(np.sum(d,axis=1))#输出[615] e=np.array([[1,2,3],[4,5,6]]) print(np.sum(e,axis=1,keepdims=True))#输出[[6][15]]

上述代码中,分别计算了数组a、b、c、d、e中元素的和。

版权声明

本文仅代表作者观点,不代表博信信息网立场。

热门