dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dosangyoon
2022-07-04 07:25:16 +09:00
parent dd54ff3597
commit a6c4d35f76
2 changed files with 112 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
stockpredictor/analysis/RSI.py
View File

@@ -46,46 +46,10 @@ class RSI:
따라서 과매수 구간에서는 매도 준비를, 과매도 구간에서는 매수 준비를 해야 한다.
"""
results = []
df = pd.DataFrame()
df = df.from_dict(stock['PRICE'])
df = self.apply(df)
for i in range(len(df.rsi)):
result = {'DATE':stock['PRICE'][i]['DATE'],
'rsi': df.rsi.values[i],
'rsis': df.rsis.values[i],
'rsi_buy': 0}
# rsi가 상향이고 30을 돌파하면 매수,
if df.rsi.values[i - 1] < df.rsi.values[i]:
if df.rsi.values[i-1] <= 30 and df.rsi.values[i] > 30:
result['rsi_buy'] = 1
# rsi가 상향이고 40을 돌파하면 매수,
if df.rsi.values[i - 1] < df.rsi.values[i]:
if df.rsi.values[i-1] <= 50 and df.rsi.values[i] > 50:
result['rsi_buy'] = 1
# rsi가 상향이고 70을 돌파하면 단기매수,
if df.rsi.values[i - 1] < df.rsi.values[i]:
if df.rsi.values[i-1] <= 70 and df.rsi.values[i] > 70:
result['rsi_buy'] = 1
# rsi가 하향이고 70이하로 떨어지면 매도,
if df.rsi.values[i - 1] > df.rsi.values[i]:
if df.rsi.values[i-1] > 70 and df.rsi.values[i] <= 70:
result['rsi_buy'] = -1
# rsi가 하향이고 50이하로 떨어지면 매도,
if df.rsi.values[i - 1] > df.rsi.values[i]:
if df.rsi.values[i-1] > 50 and df.rsi.values[i] <= 50:
result['rsi_buy'] = -1
# rsi가 하향이고 30이하로 떨어지면 단기매도,
if df.rsi.values[i - 1] > df.rsi.values[i]:
if df.rsi.values[i-1] > 30 and df.rsi.values[i] <= 30:
result['rsi_buy'] = -1
results.append(result)
return results
stock['PRICE'][i]['rsi'] = df.rsi.values[i]
stock['PRICE'][i]['rsis'] = df.rsis.values[i]