Python cumsums和cumprod函数

Posted 2020-08-12

 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