Spaces:
Sleeping
Sleeping
| import pandas as pd | |
| import pandas_ta as ta | |
| class TechnicalAnalysis: | |
| def __init__(self, csv_file): | |
| # Load the CSV data | |
| self.data = pd.read_csv(csv_file) | |
| self.data['Date'] = pd.to_datetime(self.data['Date']) | |
| self.data.set_index('Date', inplace=True) | |
| def moving_average(self, length): | |
| # Calculate moving average | |
| self.data[f'MA_{length}'] = ta.sma(self.data['Close'], length=length) | |
| def rsi(self, length=14): | |
| # Calculate RSI | |
| self.data[f'RSI_{length}'] = ta.rsi(self.data['Close'], length=length) | |
| def bollinger_bands(self, length=20, std=2): | |
| # Calculate Bollinger Bands | |
| bbands = ta.bbands(self.data['Close'], length=length, std=std) | |
| self.data['BB_upper'], self.data['BB_middle'], self.data['BB_lower'] = bbands.iloc[:, 0], bbands.iloc[:, 1], bbands.iloc[:, 2] | |
| def macd(self, fast=12, slow=26, signal=9): | |
| # Calculate MACD | |
| macd = ta.macd(self.data['Close'], fast=fast, slow=slow, signal=signal) | |
| self.data['MACD'], self.data['MACD_Signal'], self.data['MACD_Hist'] = macd.iloc[:, 0], macd.iloc[:, 1], macd.iloc[:, 2] | |
| def ultimate_oscillator(self, fast=7, medium=14, slow=28): | |
| # Calculate Ultimate Oscillator | |
| self.data['Ultimate_Oscillator'] = ta.uo(self.data['High'], self.data['Low'], self.data['Close'], fast=fast, medium=medium, slow=slow) | |
| def adx(self, length=14): | |
| # Calculate ADX | |
| self.data['ADX'], self.data['ADX_DMP'], self.data['ADX_DMN'] = ta.adx(self.data['High'], self.data['Low'], self.data['Close'], length=length) | |
| def get_data(self): | |
| return self.data |