cumsum函数是Python中numpy库的一个函数,用于计算数组元素的累积和,这个函数在数据分析和科学计算中非常有用,可以帮助我们快速计算一系列数值的累计和。
(图片来源网络,侵删)我们需要导入numpy库,然后创建一个数组,我们可以创建一个包含1到5的数组:
import numpy as np arr = np.array([1, 2, 3, 4, 5])
接下来,我们可以使用cumsum函数来计算数组元素的累积和:
cumulative_sum = np.cumsum(arr)
现在,cumulative_sum
变量包含了一个新的数组,其中每个元素都是原始数组中对应位置及其之前所有元素的和,在这个例子中,cumulative_sum
的值应该是[1, 3, 6, 10, 15]
。
cumsum函数还接受一个可选参数axis
,用于指定沿哪个轴计算累积和,默认情况下,axis
为None,表示对整个数组进行操作,如果我们有一个二维数组,可以通过设置axis
为0或1来分别计算行或列的累积和。
我们创建一个二维数组:
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
计算行累积和(axis=1
):
row_cumulative_sum = np.cumsum(arr_2d, axis=1)
计算列累积和(axis=0
):
column_cumulative_sum = np.cumsum(arr_2d, axis=0)
cumsum函数是一个非常实用的工具,可以帮助我们快速计算数组元素的累积和,通过设置不同的axis
参数,我们还可以在多维数组上实现更复杂的累积和计算。
最新评论
本站CDN与莫名CDN同款、亚太CDN、速度还不错,值得推荐。
感谢推荐我们公司产品、有什么活动会第一时间公布!
我在用这类站群服务器、还可以. 用很多年了。