Numpy中的累加和和累乘

在本文中，我们将介绍NumPy中的累加和和累乘操作。

阅读更多：Numpy 教程

累加和

累加和是指将一个数组中的元素从第一个元素开始逐个相加的结果。在NumPy中，可以使用cumsum函数来实现这一操作。cumsum函数返回的是一个新的数组，包含原数组中逐个元素的累加和。

下面是一个例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.cumsum(a)

print(b)

输出结果为：

array([ 1,  3,  6, 10, 15])

在这个例子中，我们定义了一个长度为5的数组a，然后使用cumsum函数计算了它的累加和，并将结果存储在数组b中。可以看到，数组b中的第一个元素是1（即a的第一个元素），而第二个元素是1+2=3（即a的前两个元素之和），以此类推，直到最后一个元素为1+2+3+4+5=15（即a中所有元素之和）。

累乘

累乘是指将一个数组中的元素从第一个元素开始逐个相乘的结果。在NumPy中，可以使用cumprod函数来实现这一操作。cumprod函数返回的是一个新的数组，包含原数组中逐个元素的累积积。

下面是一个例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.cumprod(a)

print(b)

输出结果为：

array([  1,   2,   6,  24, 120])

在这个例子中，我们定义了一个长度为5的数组a，然后使用cumprod函数计算了它的累积积，并将结果存储在数组b中。可以看到，数组b中的第一个元素是1（即a的第一个元素），而第二个元素是1×2=2（即a的前两个元素之积），以此类推，直到最后一个元素为1×2×3×4×5=120（即a中所有元素之积）。

使用axis参数

cumsum和cumprod函数都支持使用axis参数来指定行或列的累加和或累积积。默认情况下，这些函数会把整个数组看作一个一维数组进行操作，但是我们也可以在指定的维度上执行操作。

下面是一个例子：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.cumsum(a, axis=0)

print(b)

输出结果为：

array([[1, 2],
       [4, 6]])

在这个例子中，我们定义了一个2×2的二维数组a，然后使用cumsum函数在第一维上计算了它的累加和，并将结果存储在数组b中。可以看到，数组b的第一行就是a的第一行，而第二行则是a的两行之和（即第一行与第二行之和）。

总结

本文介绍了NumPy中的累加和和累乘操作，以及如何在指定维度上执行操作。这些操作对于对数组中的元素进行统计非常有用，尤其是在数学、统计学、概率和物理等领域中。如果您需要从整个数组中提取某些信息或进行累加运算，请务必了解和使用这些函数。