calculate exponential moving average in python

by

EDIT:
It seems that mov_average_expw() function from scikits.timeseries.lib.moving_funcs submodule from SciKits (add-on toolkits that complement SciPy) better suits the wording of your question.

To calculate an exponential smoothing of your data with a smoothing factor alpha (it is (1 - alpha) in Wikipedia’s terms):

>>> alpha = 0.5
>>> assert 0 < alpha <= 1.0
>>> av = sum(alpha**n.days * iq 
...     for n, iq in map(lambda (day, iq), today=max(days): (today-day, iq), 
...         sorted(zip(days, IQ), key=lambda p: p[0], reverse=True)))
95.0

The above is not pretty, so let’s refactor it a bit:

from collections import namedtuple
from operator    import itemgetter

def smooth(iq_data, alpha=1, today=None):
    """Perform exponential smoothing with factor `alpha`.

    Time period is a day.
    Each time period the value of `iq` drops `alpha` times.
    The most recent data is the most valuable one.
    """
    assert 0 < alpha <= 1

    if alpha == 1: # no smoothing
        return sum(map(itemgetter(1), iq_data))

    if today is None:
        today = max(map(itemgetter(0), iq_data))

    return sum(alpha**((today - date).days) * iq for date, iq in iq_data)

IQData = namedtuple("IQData", "date iq")

if __name__ == "__main__":
    from datetime import date

    days = [date(2008,1,1), date(2008,1,2), date(2008,1,7)]
    IQ = [110, 105, 90]
    iqdata = list(map(IQData, days, IQ))
    print("\n".join(map(str, iqdata)))

    print(smooth(iqdata, alpha=0.5))

Example:

$ python26 smooth.py
IQData(date=datetime.date(2008, 1, 1), iq=110)
IQData(date=datetime.date(2008, 1, 2), iq=105)
IQData(date=datetime.date(2008, 1, 7), iq=90)
95.0

More Related Contents:

  1. How to implement band-pass Butterworth filter with Scipy.signal.butter
  2. Finding the average of a list
  3. How to smooth a curve in the right way?
  4. Analyze audio using Fast Fourier Transform
  5. Finding moving average from data points in Python
  6. Python baseline correction library
  7. Creating lowpass filter in SciPy – understanding methods and units
  8. Averaging over every n elements of a numpy array
  9. scipy signal find_peaks_cwt not finding the peaks accurately?
  10. FFT-based 2D convolution and correlation in Python
  11. find time shift between two similar waveforms
  12. 2D Convolution in Python similar to Matlab’s conv2
  13. Average time for datetime list
  14. Can I convert spectrograms generated with librosa back to audio?
  15. How to smooth a curve for a dataset
  16. Digitizing an analog signal
  17. Calculating the averages for each KEY in a Pairwise (K,V) RDD in Spark with Python
  18. Generator expressions vs. list comprehensions
  19. Why am I seeing “TypeError: string indices must be integers”?
  20. Nested classes’ scope?
  21. How can you set class attributes from variable arguments (kwargs) in python
  22. 2D list has weird behavor when trying to modify a single value [duplicate]
  23. How to use QScrollArea to make scrollbars appear
  24. How to test a function with input call?
  25. How do I get the object if it exists, or None if it does not exist in Django?
  26. Using a dict to translate numbers to letters in python
  27. Is there any case where len(someObj) does not call someObj’s __len__ function?
  28. Split string at every position where an upper-case word starts
  29. Connection Timeout with Elasticsearch
  30. name ‘times’ is used prior to global declaration – But IT IS declared

Leave a Comment