""" Simple Moving Average (SMA) indicator implementation. """ import pandas as pd from ..base import BaseIndicator class SMAIndicator(BaseIndicator): """ Simple Moving Average (SMA) technical indicator. Calculates the unweighted mean of previous n periods. Handles sparse data appropriately without interpolation. """ def calculate(self, df: pd.DataFrame, period: int = 20, price_column: str = 'close') -> pd.DataFrame: """ Calculate Simple Moving Average (SMA). Args: df: DataFrame with OHLCV data period: Number of periods for moving average (default: 20) price_column: Price column to use ('open', 'high', 'low', 'close') Returns: DataFrame with SMA values and metadata, indexed by timestamp """ # Validate input data if not self.validate_dataframe(df, period): return pd.DataFrame() try: df = df.copy() df['sma'] = df[price_column].rolling(window=period, min_periods=period).mean() # Only keep rows with valid SMA, and only 'timestamp' and 'sma' columns result_df = df.loc[df['sma'].notna(), ['timestamp', 'sma']].copy() result_df = result_df.iloc[period-1:] result_df.set_index('timestamp', inplace=True) return result_df except Exception: return pd.DataFrame()