1. 程式人生 > >numpy的cumsum ()函式

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