ec8f5514bb
- Updated all technical indicators to return pandas DataFrames instead of lists, improving consistency and usability. - Modified the `calculate` method in `TechnicalIndicators` to directly return DataFrames with relevant indicator values. - Enhanced the `data_integration.py` to utilize the new DataFrame outputs for better integration with charting. - Updated documentation to reflect the new DataFrame-centric approach, including usage examples and output structures. - Improved error handling to ensure empty DataFrames are returned when insufficient data is available. These changes streamline the indicator calculations and improve the overall architecture, aligning with project standards for maintainability and performance.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
"""
|
|
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() |