init

stock/util/LabelMaker.py

@@ -8,6 +8,7 @@ from plotly import subplots
 from hts.HTS import HTS
 from hts.BuySellChecker import BuySellChecker
 
+
 class LabelMaker (HTS):
     buySellChecker = None
 
@@ -29,10 +30,10 @@ class LabelMaker (HTS):
         for i in range(size-60):
             min_price, min_price_c, max_price, max_price_c = 9999999, -1, 0, -1
             for c in range(i, i+60):
-                if data["close"][c] > max_price:
-                    max_price = data["close"][c]
+                if max(data["close"][c], data["close"][c]) > max_price:
+                    max_price = max(data["close"][c], data["close"][c])
                     max_price_c = c
-                if data["close"][c] < min_price:
+                if min(data["close"][c], data["close"][c]) < min_price:
                     min_price = data["close"][c]
                     min_price_c = c
 
@@ -45,7 +46,51 @@ class LabelMaker (HTS):
 
         return bsLine, data
 
-    def draw(self, stock_code, given_day, data, bsLine):
+    def draw_simple(self, stock_code, given_day, data, bsLine):
+        buy_line = bsLine['buy']
+        sell_line = bsLine['sell']
+
+        # 그래프 설정을 위한 변수를 생성한다.
+        data = data.astype({'open': 'int',
+                            'high': 'int',
+                            'low': 'int',
+                            'close': 'int'
+                            })
+
+        buy_colors = []
+        for i in range(len(buy_line)):
+            if buy_line[i] < 0:
+                buy_colors.append("#ffffff")
+                buy_line[i] = nan
+            else:
+                buy_colors.append("#ff00ff")
+        sell_colors = []
+        for i in range(len(sell_line)):
+            if sell_line[i] < 0:
+                sell_colors.append("#ffffff")
+                sell_line[i] = nan
+            else:
+                sell_colors.append("#00ced1")
+
+        # 그래프를 설정한다.
+        buy_check = go.Scatter(x=data['date'], y=buy_line, mode='markers', name="buy", marker=dict(size=14, color=buy_colors, line_width=0))
+        sell_check = go.Scatter(x=data['date'], y=sell_line, mode='markers', name="sell", marker=dict(size=14, color=sell_colors, line_width=0))
+        candle_stick = go.Candlestick(x=data['date'], open=data['open'], high=data['high'], low=data['low'], close=data['close'], increasing_line_color='red', decreasing_line_color='blue')
+        candle_data = [candle_stick, buy_check, sell_check]
+
+        df = pd.DataFrame(bsLine)
+        df = df.fillna(-1)
+        buy_count = len(df.loc[df["buy"] > 0])
+        sell_count = len(df.loc[df["sell"] > 0])
+
+        self.fig = go.FigureWidget(data=candle_data)
+        self.buy_data = self.fig.data[1]
+        self.fig.update_layout(height=1000, title=stock_code + "_" + given_day + "_" + str(buy_count) + "," + str(sell_count))
+        self.fig.show()
+
+        return
+
+    def draw_detail(self, stock_code, given_day, data, bsLine):
         buy_line = bsLine['buy']
         sell_line = bsLine['sell']
 
@@ -114,17 +159,6 @@ class LabelMaker (HTS):
         stochastic_data = [slow_k_line, slow_d_line]
         rsi_data = [rsi_line, rsis_line]
 
-        """
-        fig = go.Figure(data=candle_data)
-        df = pd.DataFrame(bsLine)
-        df = df.fillna(-1)
-        buy_count = len(df.loc[df["buy"] > 0])
-        sell_count = len(df.loc[df["sell"] > 0])
-
-        fig.update_layout(height=1000, title=stock_code + "_" + given_day + "_" + str(buy_count)+","+str(sell_count))
-        fig.show()
-        """
-
         fig = subplots.make_subplots(rows=5, cols=1, subplot_titles=('캔들', "거래량", "MACD", "스토캐스틱", "RSI"))
         for trace in candle_data:
             fig.append_trace(trace, 1, 1)
@@ -147,7 +181,6 @@ class LabelMaker (HTS):
         fig.update_layout(height=3000, title=stock_code + "_" + given_day + "_" + str(buy_count)+","+str(sell_count))
         fig.show()
 
-
         return
 
     def writeLabelFile(self, bsLine, data, ymd):
@@ -174,7 +207,8 @@ class LabelMaker (HTS):
         bsLine, data = self.checkTransaction(data)
 
         self.writeLabelFile(bsLine, data, ymd)
-        self.draw(stock_code, ymd, data, bsLine)
+        self.draw_simple(stock_code, ymd, data, bsLine)
+        self.draw_detail(stock_code, ymd, data, bsLine)
         return
 
 if __name__ == "__main__":
@@ -186,4 +220,4 @@ if __name__ == "__main__":
     stock_code = "252670"
     #stock_code = "122630"
 
-    labelMaker.makeCandidate(stock_code, "20220727")
+    labelMaker.makeCandidate(stock_code, "20220802")