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

1 vote325 views2 answers
Pandas save dataframe to csv in python
Python
0 vote313 views1 answer
Install python 3.8 on wsl
Python
0 vote299 views1 answer
UTC to ISO 8601 with Local TimeZone information without microsecond (Python 3):
Python
0 vote331 views1 answer
Multi-line comments Python Example
Python
0 vote324 views1 answer
Pyqt5 date-edity
Python
35 votes311 views6 answers
Install matplotlib
Python
1 vote333 views1 answer
Convert seconds to hours python
Python
51 votes341 views6 answers
Drop a column in pandas
Python
0 vote345 views1 answer
Distribute a local Python collection to form an RDD
Python
11 votes329 views6 answers
Curl to python
Python