diff --git a/stockpredictor/analysis/RSI.py b/stockpredictor/analysis/RSI.py index 0dca0e3..a0db0f9 100644 --- a/stockpredictor/analysis/RSI.py +++ b/stockpredictor/analysis/RSI.py @@ -1,9 +1,6 @@ import pandas as pd from stockpredictor.analysis.Common import Common -from plotly import tools, subplots import numpy as np -import plotly.graph_objs as go -import plotly.io as po # [청송촌놈] 파생을 알아야 시장이 보인다. 청송이 종목 고르는법! https://www.youtube.com/watch?v=weABtgZDeGg # 6. Pandas와 Plotly를 이용한 MACD 차트 그리기 https://excelsior-cjh.tistory.com/110 @@ -20,27 +17,6 @@ class RSI: self.common = Common() return - def draw(self, stock): - item_name = stock["NAME"] - item_code = stock["CODE"] - - df = pd.DataFrame(stock["PRICE"]) - rsi = go.Scatter(x=df.DATE, y=df['rsi'], name="RSI") - signal = go.Scatter(x=df.DATE, y=df['rsis'], name="RSI Signal") - data = [rsi, signal] - - layout = go.Layout(title='{} RSI 그래프'.format(item_name)) - fig = subplots.make_subplots(rows=2, cols=1, shared_xaxes=True) - - for trace in data: - fig.append_trace(trace, 1,1) - - fig = go.Figure(data=data, layout=layout) - - path = "/Users/dsyoon/workspace/StockPredictor/resources/analysis/html" - po.write_html(fig, file=path + "/rsi" + item_code+'.html', auto_open=False) - return fig - def apply(sefl, df, period=14): # df.diff를 통해 (기준일 종가 - 기준일 전일 종가)를 계산하여 0보다 크면 증가분을 감소했으면 0을 넣어줌 U = np.where(df.close.diff(1) > 0, df.close.diff(1), 0)