首页 > 代码库 > Python cumsums和cumprod函数

Python cumsums和cumprod函数

 1 >>>a = np.array([1,2,3],[4,5,6]]) 2 >>>a 3 array([[1,2,3], 4        [4,5,6]]) 5 >>>a.cumsum(0) 6 array([[1,2,3], 7        [5,7,9]]) 8 >>>a.cumprod(1) 9 array([[1,2,6],10        [4,20,120]])

这两个函数中难点就是其中参数0,1

 

其中0代表列的计算,1代表行的计算,即对列和行分别累积求和、 积。

而且其结果不聚合,产生的是中间数组。

 

详情可参考:http://docs.scipy.org/doc/numpy/reference/generated/numpy.cumsum.html

Python cumsums和cumprod函数