numpy sum函数的用法是什么

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中元素的和。