Numpy rolling average

Moving average numpy

def moving_average(a, n=3) :
    ret = np.cumsum(a, dtype=float)
    ret[n:] = ret[n:] - ret[:-n]
    return ret[n - 1:] / n

>>> a = np.arange(20)
>>> moving_average(a)
array([  1.,   2.,   3.,   4.,   5.,   6.,   7.,   8.,   9.,  10.,  11.,
        12.,  13.,  14.,  15.,  16.,  17.,  18.])
>>> moving_average(a, n=4)
array([  1.5,   2.5,   3.5,   4.5,   5.5,   6.5,   7.5,   8.5,   9.5,
        10.5,  11.5,  12.5,  13.5,  14.5,  15.5,  16.5,  17.5])

Source: stackoverflow.com

Numpy rolling average

import numpy as np

def moving_average(x, w):
    return np.convolve(x, np.ones(w), 'valid') / w

Source: stackoverflow.com

Similar codes

6 votes351 views2 answers
Python square a number
Python
0 vote298 views0 answer
Python
0 vote349 views1 answer
Reading a cell from another cell in colab
Python
24 votes349 views6 answers
Display all rows pandas
Python
0 vote329 views1 answer
Check if number is divisible without remainder python
Python
30 votes469 views3 answers
Django create superuser from script
Python Django
36 votes342 views6 answers
Requirement.txt for python
Python
-1 vote334 views1 answer
Check runtime python
Python
0 vote352 views1 answer
Unpacking using List in Python
Python
74 votes323 views6 answers
Append dictionary python
Python