How to Calculate Moving Averages in Python

Spread the love

Introduction

Moving averages are widely used in data analysis and particularly in time-series analysis. It is a simple yet powerful technique that is commonly used in stock price analysis, sensor data smoothing, trend analysis, and more. In this article, we will delve into the various types of moving averages and provide comprehensive insights on how to calculate them using Python.

Table of Contents

  1. Understanding Moving Averages
  2. Install Required Libraries
  3. Simple Moving Average
  4. Weighted Moving Average
  5. Exponential Moving Average
  6. Cumulative Moving Average
  7. Using Libraries to Calculate Moving Averages
  8. Practical Example with Stock Data
  9. Visualizing Moving Averages
  10. Applications of Moving Averages
  11. Conclusion

1. Understanding Moving Averages

Moving average, as the name suggests, involves calculating the average of a set of data points over a specific period. It is ‘moving’ because as new data points are added, the oldest data points are dropped, and the average recalculates. There are different types of moving averages including Simple Moving Average, Weighted Moving Average, Exponential Moving Average, and Cumulative Moving Average.

2. Install Required Libraries

Install the required libraries.

pip install numpy pandas matplotlib yfinance

3. Simple Moving Average (SMA)

Simple Moving Average (SMA) is the most basic type. It calculates the average of a selected range of prices, usually closing prices, by the number of periods in that range.

import numpy as np

def simple_moving_average(data, window_size):
    return np.convolve(data, np.ones(window_size), 'valid') / window_size

# Example
data = [2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
window_size = 3
sma = simple_moving_average(data, window_size)
print(sma)

4. Weighted Moving Average (WMA)

Unlike SMA, the Weighted Moving Average (WMA) assigns different weights to the data points based on their age with newer data points carrying more weight.

def weighted_moving_average(data, weights):
    return np.convolve(data, weights, 'valid') / sum(weights)

# Example
data = [2, 3, 4, 5, 6]
weights = [0.1, 0.2, 0.3, 0.4]
wma = weighted_moving_average(data, weights)
print(wma)

5. Exponential Moving Average (EMA)

Exponential Moving Average gives more weight to the recent prices and as a result, it can be more reactive to recent price changes compared to SMA.

def exponential_moving_average(data, alpha):
    ema = []
    for i, x in enumerate(data):
        if i == 0:
            ema.append(x)
        else:
            ema.append(alpha * x + (1 - alpha) * ema[-1])
    return ema

# Example
data = [2, 3, 4, 5, 6]
alpha = 0.1
ema = exponential_moving_average(data, alpha)
print(ema)

6. Cumulative Moving Average (CMA)

Cumulative Moving Average considers all prior data points rather than a sliding window. It is usually not practical for large datasets but can be useful in certain scenarios.

def cumulative_moving_average(data):
    return np.cumsum(data) / (np.arange(len(data)) + 1)

# Example
data = [2, 3, 4, 5, 6]
cma = cumulative_moving_average(data)
print(cma)

7. Using Libraries to Calculate Moving Averages

7.1 Using Pandas

Pandas is a powerful data manipulation library. We can calculate moving averages with ease using its rolling method.

import pandas as pd

data = pd.Series([2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
sma = data.rolling(window=3).mean()
print(sma)

8. Practical Example with Stock Data

Let’s work with real-world stock market data to calculate moving averages.

import yfinance as yf
import pandas as pd

# Fetch data
data = yf.download("AAPL", start="2023-01-01", end="2023-06-01")

# Calculate SMA and EMA
data['SMA'] = data['Close'].rolling(window=20).mean()
data['EMA'] = data['Close'].ewm(span=20).mean()

9. Visualizing Moving Averages

Visualizing data is a crucial step in data analysis. Let’s plot the stock prices along with the moving averages.

import matplotlib.pyplot as plt

plt.figure(figsize=(12,6))
plt.plot(data['Close'], label='Stock Price')
plt.plot(data['SMA'], label='20-Day Simple Moving Average')
plt.plot(data['EMA'], label='20-Day Exponential Moving Average')
plt.legend(loc='upper left')
plt.title('AAPL Stock Price and Moving Averages')
plt.xlabel('Date')
plt.ylabel('Stock Price')
plt.show()

10. Applications of Moving Averages

  1. Trend Analysis: Moving averages can help to visually depict the trend direction of a dataset.
  2. Stock Market Analysis: They are widely used to smooth out price data to help traders make better decisions.
  3. Sensor Data Smoothing: Used for filtering out noise from sensors in various engineering applications.
  4. Algorithmic Trading: Many trading strategies are based on moving averages crossovers.

11. Conclusion

In this article, we have covered the concept of moving averages and how to calculate them in Python using various methods including Simple Moving Average, Weighted Moving Average, Exponential Moving Average, and Cumulative Moving Average. We also saw how to work with real-world stock data and visualize moving averages using Matplotlib. Moving averages are a fundamental tool in time-series analysis and knowing how to use them effectively can be incredibly valuable for data analysis, trading, and making data-driven decisions.

Leave a Reply