From 86280c0ea5e32f28955a273b087ae93f58ee4cea Mon Sep 17 00:00:00 2001 From: dsyoon Date: Thu, 25 Jan 2024 09:17:48 +0900 Subject: [PATCH] init --- stock/analysis/AnalyzerSqlite.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/stock/analysis/AnalyzerSqlite.py b/stock/analysis/AnalyzerSqlite.py index 200b2c1..c0e5af4 100644 --- a/stock/analysis/AnalyzerSqlite.py +++ b/stock/analysis/AnalyzerSqlite.py @@ -725,13 +725,14 @@ class AnalyzerSqlite: self.rsi.analyze(stock) self.macd.analyze(stock) + close_list = [price['close'] for price in stock['PRICE']] # 트렌드: 실시간 정상 작동 안됨 - size = int(len(stock['PRICE']) / 20) + size = int(len(close_list) / 20) pos = round(size / 2) - close_temp = stock["close"] + [stock["close"][-1]] * pos + close_temp = close_list + [close_list[-1]] * pos decomposition_results = seasonal_decompose(close_temp, model='multiplicative', period=size) trend = decomposition_results.trend[:-pos] - trend_df = pd.DataFrame(trend).fillna(stock["close"][0]) + trend_df = pd.DataFrame(trend).fillna(close_list[0]) sorted_stock = sorted(stock["PRICE"], key=lambda x: x['ymd'], reverse=True) for price in sorted_stock: