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

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


咨詢
建站咨詢
