Create indicators/sma.py

indicators/sma.py

@@ -0,0 +1,21 @@
+import pandas as pd
+
+def calculate_sma(data, column='close', period=21):
+    """
+    Calculates the Simple Moving Average (SMA) for a specified column in the data DataFrame.
+
+    Args:
+        data (pd.DataFrame): The input DataFrame containing the price data.
+        column (str): The name of the column to calculate the SMA on. Default is 'close'.
+        period (int): The period over which to calculate the SMA. Default is 21.
+
+    Returns:
+        pd.Series: A pandas Series containing the SMA for the specified period.
+    """
+    sma = data[column].rolling(window=period, min_periods=1).mean()
+    return sma
+
+# Example usage:
+# Assuming `data` is a pandas DataFrame with a 'close' column containing closing prices:
+# sma21 = calculate_sma(data, column='close', period=21)
+# data['SMA_21'] = sma21