numpy的cumsum （）函數

cumsum是matlab中一個函數，一般用於計算一個數組各行的累加值，函數用法是B = cumsum(A,dim)，或B = cumsum(A)。

 

函數功能

調用格式及說明

格式一：B = cumsum(A)

這種用法返回數組不一樣維數的累加和。

爲了便於接下來的敘述，解釋一下matlab中矩陣、數組、向量的概念： [1] 

首先，matlab的是矩陣實驗室的意思。也就是說matlab中的數據都被視爲矩陣。

數組就是一個一行n列的矩陣，向量就是一個n行一列的矩陣。

若是A是一個向量， cumsum(A) 返回一個向量，該向量中第m行的元素是A中第1行到第m行的全部元素累加和；

若是A是一個矩陣， cumsum(A) 返回一個和A同行同列的矩陣，矩陣中第m行第n列元素是A中第1行到第m行的全部第n列元素的累加和；

若是A是一個多維數組， cumsum(A)只對A中第一個非奇異維進行計算。

格式二：B = cumsum(A,dim)

這種調用格式返回A中由標量dim所指定的維數的累加和。例如：cumsum(A,1)返回的是沿着第一維（各列）的累加和，cumsum(A,2)返回的是沿着第二維（各行）的累加和。

具體用法參考程序示例或matlab的幫助文檔。

Aaa =

1 2 3

4 5 6

>> cumsum(Aaa,1)

ans =

1 2 3

5 7 9

>> cumsum(Aaa,2)

ans =

1 3 6

4 9 15