Python Running cumulative sum with a given window

前端 未结 4 866
离开以前
离开以前 2020-12-10 19:09

What I want to do is generate a numpy array that is the cumulative sum of another numpy array given a certain window.

For example, given an array [1,2,3,4,5,6

4条回答
  •  醉酒成梦
    2020-12-10 19:35

    Here is perhaps a simpler answer, based on subtracting shifted cumsums.

    >>> a = np.array([1,2,3,4,5,6,7,8,9,10,11,12])
    >>> b = a.cumsum()
    >>> b[3:] = b[3:] - b[:-3]
    >>> b
    array([ 1,  3,  6,  9, 12, 15, 18, 21, 24, 27, 30, 33])
    

提交回复
热议问题