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函數的具體例子。