Python实现cumsum函数

引言

累计求和是在数学和统计学中一个非常重要的概念。在Python中，我们可以使用cumsum函数来完成累计求和的操作。

什么是cumsum函数？

cumsum函数是numpy库中的一个函数，用于计算输入数组的累计和（cumulative sum）。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
cum_sum_arr = np.cumsum(arr)
print(cum_sum_arr)  # [1 3 6 10 15]

cumsum函数的语法

numpy.cumsum(arr, axis=None, dtype=None, out=None)

arr：要累加的数组。

axis：累加的轴，如果未提供此参数，则平铺数组。

dtype：数据类型。

out：用于存储累加结果的输出数组。

cumsum函数的例子

例子1：用cumsum函数计算一维数组的累计和

在这个例子中，我们使用cumsum函数计算一维数组的累计和。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
cum_sum_arr = np.cumsum(arr)
print(cum_sum_arr)  # [1 3 6 10 15]

例子2：用cumsum函数计算矩阵的累计和

在这个例子中，我们使用cumsum函数计算一个矩阵的累计和。

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
cum_sum_arr = np.cumsum(arr, axis=0)  # 对行求累加
print(cum_sum_arr)  # [[ 1  2] [ 4  6] [ 9 12]]

cum_sum_arr = np.cumsum(arr, axis=1)  # 对列求累加
print(cum_sum_arr)  # [[ 1  3] [ 3  7] [ 5 11]]

例子3：用cumsum函数计算带有单位的数组的累计和

在这个例子中，我们使用cumsum函数计算带有测量单位的数组的累计和。

import numpy as np

arr = np.array([1, 2, 3, 4, 5]) * 2
cum_sum_arr = np.cumsum(arr, dtype=float)
print(cum_sum_arr)  # [  2.   6.  12.  20.  30.]

总结

累计求和是一个重要的统计量，使用cumsum函数可以方便地计算输入数组的累计和。在本文中，我们学习了cumsum函数的语法和用法，并给出了几个使用numpy库中cumsum函数的具体例子。