Python中的 numpy.sum()

numpy.sum()是Python中的NumPy库中的一种操作，用于计算给定数组中的元素的总和。它对于对于需要对数组中所有元素进行求和的任务特别有用。

用法

下面是numpy.sum()的基本语法：

numpy.sum(a, axis=None, dtype=None, out=None, keepdims=<no value>)

其中:

• a - 要求和的输入数组
• axis - 沿哪个轴计算。如果没有给定，就计算所有元素的总和。
• dtype - 返回数组的数据类型
• out - 结果的输出数组
• keepdims - 是否保留每个维度的输出尺寸。如果设置为True，则输出的每个维度将与输入的大小相同。否则，将会丢弃一些维度。

代码示例

下面是一个简单的代码示例，展示了如何使用numpy.sum()来计算一个一维数组的元素总和：

import numpy as np

# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5])

# 计算元素总和
result = np.sum(a)

print(result)

输出：

15

下面是一个更复杂的例子，演示了如何使用numpy.sum()来计算一个二维数组沿轴0和轴1的元素总和：

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 计算沿轴0的元素总和
result_axis0 = np.sum(a, axis=0)

# 计算沿轴1的元素总和
result_axis1 = np.sum(a, axis=1)

print(result_axis0)
print(result_axis1)

输出：

[12 15 18]
[ 6 15 24]

此外，numpy.sum()还有很多其他功能，例如按位求和、指定数据类型等等。

结束语

numpy.sum()是一个非常实用的函数，可以对数组中的元素进行快速求和。对于处理大型数据集和数字计算的任务来说，这个函数是一个强大的工具。