阅读量: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,则结果是一个简化的数组。
返回值: 返回一个数组或者标量,表示数组中元素的和。
示例:
import numpy as np 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)) # 输出[5 7 9] d = np.array([[1, 2, 3], [4, 5, 6]]) print(np.sum(d, axis=1)) # 输出[ 6 15] e = np.array([[1, 2, 3], [4, 5, 6]]) print(np.sum(e, axis=1, keepdims=True)) # 输出[[ 6] [15]]
上述代码中,分别计算了数组a、b、c、d、e中元素的和。