numpy求和

什么是numpy？

NumPy是一个用于科学计算的Python库，提供了快速高效的多维数组对象ndarray。除了数组对象，NumPy还提供了许多用于对数组进行操作的函数和方法。其中包括对数组进行求和的函数。

numpy求和函数

在NumPy中，求和是一种常用的操作，可以对数组的元素进行求和。NumPy中提供了不同的函数可以用来对数组进行求和，常用的有numpy.sum()函数和数组对象的sum()方法。

numpy.sum()函数

numpy.sum()函数可以用来对数组中的所有元素进行求和。它的语法如下：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = np.sum(arr)
print(result)

运行结果：

数组对象的sum()方法

数组对象也提供了sum()方法，可以直接对数组对象调用该方法进行求和。它的语法如下：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = arr.sum()
print(result)

运行结果：

对多维数组求和

除了对一维数组求和，NumPy还可以对多维数组进行求和。对于多维数组，我们可以指定求和的维度。

对二维数组求和

对于二维数组，我们可以指定沿着某个维度进行求和。例如，对于一个二维数组，我们可以沿着行或列进行求和。示例代码如下：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
result1 = np.sum(arr, axis=0)  # 沿着列求和
result2 = arr.sum(axis=1)      # 沿着行求和
print(result1)
print(result2)

运行结果：

[5 7 9]
[ 6 15]

对三维数组求和

对于三维数组，我们同样可以指定沿着某个维度进行求和。示例代码如下：

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
result1 = np.sum(arr, axis=0)  # 沿着第一个维度求和
result2 = arr.sum(axis=1)      # 沿着第二个维度求和
print(result1)
print(result2)

运行结果：