新聞中心
cumsum函數(shù)是Python中numpy庫(kù)的一個(gè)函數(shù),用于計(jì)算數(shù)組元素的累積和,這個(gè)函數(shù)在數(shù)據(jù)分析和科學(xué)計(jì)算中非常有用,可以幫助我們快速計(jì)算一系列數(shù)值的累計(jì)和。

我們需要導(dǎo)入numpy庫(kù),然后創(chuàng)建一個(gè)數(shù)組,我們可以創(chuàng)建一個(gè)包含1到5的數(shù)組:
import numpy as np arr = np.array([1, 2, 3, 4, 5])
接下來(lái),我們可以使用cumsum函數(shù)來(lái)計(jì)算數(shù)組元素的累積和:
cumulative_sum = np.cumsum(arr)
現(xiàn)在,cumulative_sum變量包含了一個(gè)新的數(shù)組,其中每個(gè)元素都是原始數(shù)組中對(duì)應(yīng)位置及其之前所有元素的和,在這個(gè)例子中,cumulative_sum的值應(yīng)該是[1, 3, 6, 10, 15]。
cumsum函數(shù)還接受一個(gè)可選參數(shù)axis,用于指定沿哪個(gè)軸計(jì)算累積和,默認(rèn)情況下,axis為None,表示對(duì)整個(gè)數(shù)組進(jìn)行操作,如果我們有一個(gè)二維數(shù)組,可以通過(guò)設(shè)置axis為0或1來(lái)分別計(jì)算行或列的累積和。
我們創(chuàng)建一個(gè)二維數(shù)組:
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
計(jì)算行累積和(axis=1):
row_cumulative_sum = np.cumsum(arr_2d, axis=1)
計(jì)算列累積和(axis=0):
column_cumulative_sum = np.cumsum(arr_2d, axis=0)
cumsum函數(shù)是一個(gè)非常實(shí)用的工具,可以幫助我們快速計(jì)算數(shù)組元素的累積和,通過(guò)設(shè)置不同的axis參數(shù),我們還可以在多維數(shù)組上實(shí)現(xiàn)更復(fù)雜的累積和計(jì)算。
文章題目:cumsum函數(shù)python
轉(zhuǎn)載來(lái)于:http://www.5511xx.com/article/dpcodsh.html


咨詢
建站咨詢
