• PyQt5_股票K线形态查看工具

    K线形态，诸如大阳线、螺旋桨、穿头破脚等，代码逻辑实现后，要使用股票数据验证，这类主要依靠视觉判断的K线形态，获得的结果能及时在图表中展示出来，对代码的开发、校验、结果的准确度确认等能起到事半功倍的效果，本工具就是基于此需求所做的开发，需要的朋友可以自取，具体K线形态判断的代码实现，请查看后续博文。

    本文以“大阳线”为例进行讲解。

    目录

    效果

    计算K线形态策略代码

    工具代码

    工具使用

    数据

    效果

    计算K线形态策略代码

     前置说明：

    1. 必须以“excute_strategy”为方法名

    2. 一个参数，股票日数据文件路径

    3. 策略代码写完后，命名，保存在‘k_recognition_strategy’文件夹中，‘k_recognition_strategy’的创建位置在下面的工具代码中会介绍

    4. 将策略文件名写入主程序类StrategeMainWidget 的 self.strategyname_code_map 对应的键值对中

    1. def excute_strategy(daily_file_path):
    2.     '''
    3.     名称：大阳线
    4.     识别：K线实体长度为上一交易日价格4%以上。
    5.     前置条件：计算时间区间 2021-01-01 到 2022-01-01
    6.     :param daily_file_path: 股票日数据文件路径
    7.     :return:
    8.     '''
    9.     import pandas as pd
    10.     import os
    11.     start_date_str = '2021-01-01'
    12.     end_date_str = '2022-01-01'
    13.     df = pd.read_csv(daily_file_path,encoding='utf-8')
    14.     # 删除停牌的数据
    15.     df = df.loc[df['openPrice'] > 0].copy()
    16.     df['o_date'] = df['tradeDate']
    17.     df['o_date'] = pd.to_datetime(df['o_date'])
    18.     df = df.loc[(df['o_date'] >= start_date_str) & (df['o_date']<=end_date_str)].copy()
    19.     # 保存未复权收盘价数据
    20.     df['close'] = df['closePrice']
    21.     # 计算前复权数据
    22.     df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    23.     df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    24.     df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    25.     df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    26.  
    27.     # 开始计算
    28.     df['body_length'] = df['closePrice'] - df['openPrice']
    29.     df['signal'] = 0
    30.     df['signal_name'] = 0
    31.     df.loc[(df['body_length']>0) & (df['body_length']/df['closePrice'].shift(1)>=0.04),'signal'] = 1
    32.     df.loc[(df['body_length']>0) & (df['body_length']/df['closePrice'].shift(1)>=0.04),'signal_name'] = (df['body_length']/df['closePrice'].shift(1))*100
    33.     df = df.round({'signal_name':2})
    34.  
    35.     file_name = os.path.basename(daily_file_path)
    36.     title_str = file_name.split('.')[0]
    37.  
    38.     line_data = {
    39.         'title_str':title_str,
    40.         'whole_header':['日期','收','开','高','低'],
    41.         'whole_df':df,
    42.         'whole_pd_header':['tradeDate','closePrice','openPrice','highestPrice','lowestPrice'],
    43.         'start_date_str':start_date_str,
    44.         'end_date_str':end_date_str,
    45.         'signal_type':'line'
    46.     }
    47.     return line_data

    工具代码

    导入需要的包

    1. import sys,json,os,math,time,talib
    2. from threading import Thread
    3. import numpy as np
    4. import pandas as pd
    5. from datetime import datetime
    6. from dateutil.relativedelta import relativedelta
    7. from typing import Dict,Any,List
    8. from PyQt5 import QtCore,QtGui,QtWidgets
    9. from PyQt5.QtCore import Qt
    10. import pyqtgraph as pg
    11. import pyqtgraph.examples
    12. pg.setConfigOption('background','k')
    13. pg.setConfigOption('foreground','w')

     pyqtgraph 日期横坐标控件

    1. class RotateAxisItem(pg.AxisItem):
    2.     def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):
    3.         p.setRenderHint(p.Antialiasing,False)
    4.         p.setRenderHint(p.TextAntialiasing,True)
    5.  
    6.         ## draw long line along axis
    7.         pen,p1,p2 = axisSpec
    8.         p.setPen(pen)
    9.         p.drawLine(p1,p2)
    10.         p.translate(0.5,0)  ## resolves some damn pixel ambiguity
    11.  
    12.         ## draw ticks
    13.         for pen,p1,p2 in tickSpecs:
    14.             p.setPen(pen)
    15.             p.drawLine(p1,p2)
    16.  
    17.         ## draw all text
    18.         # if self.tickFont is not None:
    19.         #     p.setFont(self.tickFont)
    20.         p.setPen(self.pen())
    21.         for rect,flags,text in textSpecs:
    22.             # this is the important part
    23.             p.save()
    24.             p.translate(rect.x(),rect.y())
    25.             p.rotate(-30)
    26.             p.drawText(-rect.width(),rect.height(),rect.width(),rect.height(),flags,text)
    27.             # restoring the painter is *required*!!!
    28.             p.restore()

     pyqtgraph 蜡烛图控件

    1. class CandlestickItem(pg.GraphicsObject):
    2.     def __init__(self, data):
    3.         pg.GraphicsObject.__init__(self)
    4.         self.data = data  ## data must have fields: time, open, close, min, max
    5.         self.generatePicture()
    6.  
    7.     def generatePicture(self):
    8.         ## pre-computing a QPicture object allows paint() to run much more quickly,
    9.         ## rather than re-drawing the shapes every time.
    10.         self.picture = QtGui.QPicture()
    11.         p = QtGui.QPainter(self.picture)
    12.         p.setPen(pg.mkPen({'color':'w','width':1}))
    13.         w = (self.data[1][0] - self.data[0][0]) / 3.
    14.         for (t, open, close, min, max) in self.data:
    15.             if open==close and open==max and max==min:
    16.                 p.drawLine(QtCore.QPointF(t-w, min), QtCore.QPointF(t+w, max))
    17.             else:
    18.                 p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))
    19.                 if open < close:
    20.                     p.setBrush(pg.mkBrush('r'))
    21.                 else:
    22.                     p.setBrush(pg.mkBrush('g'))
    23.                 p.drawRect(QtCore.QRectF(t-w, open, w * 2, close - open))
    24.         p.end()
    25.  
    26.     def paint(self, p, *args):
    27.         p.drawPicture(0, 0, self.picture)
    28.  
    29.     def boundingRect(self):
    30.         ## boundingRect _must_ indicate the entire area that will be drawn on
    31.         ## or else we will get artifacts and possibly crashing.
    32.         ## (in this case, QPicture does all the work of computing the bouning rect for us)
    33.         return QtCore.QRectF(self.picture.boundingRect())

     分页表格控件

    1. class PageTableWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.         pass
    7.  
    8.     def init_data(self):
    9.         # 表格全数据 二维数组
    10.         self.table_full_data: List[Any] = []
    11.         # 表格右键菜单 {菜单名：索引指向}
    12.         self.table_right_menus: Dict[str, str] = {}
    13.  
    14.         self.total_page_count: int = 0
    15.         self.total_rows_count: int = 0
    16.         self.current_page: int = 1
    17.         self.single_page_rows: int = 50
    18.         pass
    19.  
    20.     def init_ui(self):
    21.         pre_page_btn = QtWidgets.QPushButton('上一页')
    22.         pre_page_btn.clicked.connect(self.pre_page_btn_clicked)
    23.         next_page_btn = QtWidgets.QPushButton('下一页')
    24.         next_page_btn.clicked.connect(self.next_page_btn_clicked)
    25.  
    26.         tip_label_0 = QtWidgets.QLabel('第')
    27.         self.witch_page_lineedit = QtWidgets.QLineEdit()
    28.         self.int_validator = QtGui.QIntValidator()
    29.         self.witch_page_lineedit.setValidator(self.int_validator)
    30.         self.witch_page_lineedit.setMaximumWidth(20)
    31.         tip_label_1 = QtWidgets.QLabel('页')
    32.         go_page_btn = QtWidgets.QPushButton('前往')
    33.         go_page_btn.clicked.connect(self.go_page_btn_clicked)
    34.         layout_witch_page = QtWidgets.QHBoxLayout()
    35.         layout_witch_page.addWidget(tip_label_0)
    36.         layout_witch_page.addWidget(self.witch_page_lineedit)
    37.         layout_witch_page.addWidget(tip_label_1)
    38.         layout_witch_page.addWidget(go_page_btn)
    39.         layout_witch_page.addStretch(1)
    40.  
    41.         layout_pagechange = QtWidgets.QHBoxLayout()
    42.         layout_pagechange.addWidget(pre_page_btn)
    43.         layout_pagechange.addWidget(next_page_btn)
    44.         layout_pagechange.addLayout(layout_witch_page)
    45.  
    46.         self.total_page_count_label = QtWidgets.QLabel(f"共0页")
    47.         self.total_rows_count_label = QtWidgets.QLabel(f"共0行")
    48.         self.current_page_label = QtWidgets.QLabel(f"当前第0页")
    49.         layout_pagestatus = QtWidgets.QHBoxLayout()
    50.         layout_pagestatus.addWidget(self.total_page_count_label)
    51.         layout_pagestatus.addWidget(self.total_rows_count_label)
    52.         layout_pagestatus.addWidget(self.current_page_label)
    53.         layout_pagestatus.addStretch(1)
    54.  
    55.         layout_top = QtWidgets.QVBoxLayout()
    56.         layout_top.addLayout(layout_pagechange)
    57.         layout_top.addLayout(layout_pagestatus)
    58.  
    59.         self.target_table = QtWidgets.QTableWidget()
    60.         self.target_table.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
    61.         self.target_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
    62.  
    63.         layout = QtWidgets.QVBoxLayout()
    64.         layout.addLayout(layout_top)
    65.         layout.addWidget(self.target_table)
    66.         self.setLayout(layout)
    67.         pass
    68.  
    69.     def set_table_init_data(self, data: Dict[str, Any]):
    70.         '''设置表头、右键菜单等初始化内容'''
    71.         headers: List[str] = data['headers']
    72.  
    73.         self.target_table.setColumnCount(len(headers))
    74.         self.target_table.setHorizontalHeaderLabels(headers)
    75.         pass
    76.  
    77.     def set_table_full_data(self, data: List[Any]):
    78.         '''表格全数据'''
    79.         self.table_full_data = data
    80.         self.total_rows_count = len(data)
    81.         self.total_page_count = math.ceil(self.total_rows_count / self.single_page_rows)
    82.         self.current_page = 1
    83.  
    84.         self.int_validator.setRange(1, self.total_page_count)
    85.  
    86.         self.total_page_count_label.setText(f"共{self.total_page_count}页")
    87.         self.total_rows_count_label.setText(f"共{self.total_rows_count}行")
    88.         self.caculate_current_show_data()
    89.         pass
    90.  
    91.     def setting_current_pagestatus_label(self):
    92.         self.current_page_label.setText(f"当前第{self.current_page}页")
    93.         pass
    94.  
    95.     def caculate_current_show_data(self):
    96.         '''计算当前要显示的数据'''
    97.         start_dot = (self.current_page - 1) * self.single_page_rows
    98.         end_dot = start_dot + self.single_page_rows
    99.         current_data = self.table_full_data[start_dot:end_dot]
    100.         self.fill_table_content(current_data)
    101.         self.setting_current_pagestatus_label()
    102.         pass
    103.  
    104.     def fill_table_content(self, data: List[Any]):
    105.         self.target_table.clearContents()
    106.         self.target_table.setRowCount(len(data))
    107.         for r_i, r_v in enumerate(data):
    108.             for c_i, c_v in enumerate(r_v):
    109.                 cell = QtWidgets.QTableWidgetItem(str(c_v))
    110.                 self.target_table.setItem(r_i, c_i, cell)
    111.                 pass
    112.             pass
    113.         self.target_table.resizeColumnsToContents()
    114.         pass
    115.  
    116.     def go_page_btn_clicked(self):
    117.         '''前往按钮点击'''
    118.         input_page = self.witch_page_lineedit.text()
    119.         input_page = input_page.strip()
    120.         if not input_page:
    121.             QtWidgets.QMessageBox.information(
    122.                 self,
    123.                 '提示',
    124.                 '请输入要跳转的页码',
    125.                 QtWidgets.QMessageBox.Yes
    126.             )
    127.             return
    128.         input_page = int(input_page)
    129.         if input_page < 0 or input_page > self.total_page_count:
    130.             QtWidgets.QMessageBox.information(
    131.                 self,
    132.                 '提示',
    133.                 '输入的页码超出范围',
    134.                 QtWidgets.QMessageBox.Yes
    135.             )
    136.             return
    137.         self.current_page = input_page
    138.         self.caculate_current_show_data()
    139.         pass
    140.  
    141.     def pre_page_btn_clicked(self):
    142.         '''上一页按钮点击'''
    143.         if self.current_page <= 1:
    144.             QtWidgets.QMessageBox.information(
    145.                 self,
    146.                 '提示',
    147.                 '已经是首页',
    148.                 QtWidgets.QMessageBox.Yes
    149.             )
    150.             return
    151.         self.current_page -= 1
    152.         self.caculate_current_show_data()
    153.         pass
    154.  
    155.     def next_page_btn_clicked(self):
    156.         '''下一页按钮点击'''
    157.         if self.current_page >= self.total_page_count:
    158.             QtWidgets.QMessageBox.information(
    159.                 self,
    160.                 '提示',
    161.                 '已经是最后一页',
    162.                 QtWidgets.QMessageBox.Yes
    163.             )
    164.             return
    165.         self.current_page += 1
    166.         self.caculate_current_show_data()
    167.         pass
    168.     pass

    K线与结果显示控件 

    1. class PyQtGraphKWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.     def init_data(self):
    7.         # https://www.sioe.cn/yingyong/yanse-rgb-16/
    8.         self.color_line = (30, 144, 255)
    9.         # 0 幽灵的白色; 1 纯黄; 2 紫红色; 3 纯绿; 4 道奇蓝
    10.         self.color_list = [(248, 248, 255), (255, 255, 0), (255, 0, 255), (0, 128, 0), (30, 144, 255)]
    11.         self.main_fixed_target_list = []  # 主体固定曲线，不能被删除
    12.         self.whole_df = None
    13.         self.whole_header = None
    14.         self.whole_pd_header = None
    15.         self.current_whole_data = None
    16.         self.current_whole_df = None
    17.         self.signal_type = None
    18.         self.duration_len = None
    19.         pass
    20.     def init_ui(self):
    21.         self.whole_duration_label = QtWidgets.QLabel('左边界~右边界')
    22.  
    23.         self.title_label = QtWidgets.QLabel('执行过程查看')
    24.         self.title_label.setAlignment(Qt.AlignCenter)
    25.         self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')
    26.  
    27.         xax = RotateAxisItem(orientation='bottom')
    28.         xax.setHeight(h=80)
    29.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    30.         self.pw.setMouseEnabled(x=True, y=True)
    31.         # self.pw.enableAutoRange(x=False,y=True)
    32.         self.pw.setAutoVisible(x=False, y=True)
    33.  
    34.         layout_right = QtWidgets.QVBoxLayout()
    35.         layout_right.addWidget(self.title_label)
    36.         layout_right.addWidget(self.whole_duration_label)
    37.         layout_right.addWidget(self.pw)
    38.         self.setLayout(layout_right)
    39.         pass
    40.  
    41.     def set_data(self, data: Dict[str, Any]):
    42.         title_str = data['title_str']
    43.         whole_header = data['whole_header']
    44.         whole_df = data['whole_df']
    45.         whole_pd_header = data['whole_pd_header']
    46.         signal_type = data['signal_type']
    47.         if signal_type == 'duration' or signal_type == 'duration_detail':
    48.             duration_len = data['duration_len']
    49.             self.duration_len = duration_len
    50.             pass
    51.  
    52.         self.whole_header = whole_header
    53.         self.whole_df = whole_df
    54.         self.whole_pd_header = whole_pd_header
    55.         self.signal_type = signal_type
    56.  
    57.         self.title_label.setText(title_str)
    58.         self.whole_duration_label.setText(f"{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")
    59.  
    60.         self.current_whole_df = self.whole_df.copy()
    61.         self.caculate_and_show_data()
    62.         pass
    63.  
    64.     def caculate_and_show_data(self):
    65.         df = self.current_whole_df.copy()
    66.         df.reset_index(inplace=True)
    67.         df['i_count'] = [i for i in range(len(df))]
    68.         tradeDate_list = df['tradeDate'].values.tolist()
    69.         x = range(len(df))
    70.         xTick_show = []
    71.         x_dur = math.ceil(len(df) / 20)
    72.         for i in range(0, len(df), x_dur):
    73.             xTick_show.append((i, tradeDate_list[i]))
    74.         if len(df) % 20 != 0:
    75.             xTick_show.append((len(df) - 1, tradeDate_list[-1]))
    76.         candle_data = []
    77.         for i, row in df.iterrows():
    78.             candle_data.append(
    79.                 (row['i_count'], row['openPrice'], row['closePrice'], row['lowestPrice'], row['highestPrice']))
    80.  
    81.         self.current_whole_data = df.loc[:, self.whole_pd_header].values.tolist()
    82.         # 开始配置显示的内容
    83.         self.pw.clear()
    84.  
    85.         xax = self.pw.getAxis('bottom')
    86.         xax.setTicks([xTick_show])
    87.  
    88.         if self.signal_type == 'line':
    89.             df_signal = df.loc[df['signal'] == 1].copy()
    90.             for i,row in df_signal.iterrows():
    91.                 signal_fiexed_target = pg.InfiniteLine(pos=(row['i_count'],0),movable=False,angle=90,pen=pg.mkPen({'color':self.color_line,'width':1}),label=str(row['signal_name']),labelOpts={'position':0.05,'color':self.color_line,'movable':True,'fill':(self.color_line[0],self.color_line[1],self.color_line[2],30)})
    92.                 self.pw.addItem(signal_fiexed_target)
    93.             pass
    94.         elif self.signal_type == 'duration':
    95.             for i,i_len in enumerate(self.duration_len):
    96.                 df_signal = df.loc[df['signal'] == i+1].copy()
    97.                 for i0, row in df_signal.iterrows():
    98.                     signal_fiexed_target = pg.LinearRegionItem([row['i_count'] - i_len + 1, row['i_count']],
    99.                                                                movable=False, brush=(
    100.                         self.color_line[0], self.color_line[1], self.color_line[2], 50))
    101.                     self.pw.addItem(signal_fiexed_target)
    102.                 pass
    103.         elif self.signal_type == 'duration_detail':
    104.             df['param_0'] = df['signal']-df['signal'].shift(1)
    105.             df['param_1'] = df['signal']-df['signal'].shift(-1)
    106.             df_start = df.loc[df['param_0']==1].copy()
    107.             df_end = df.loc[df['param_1']==1].copy()
    108.             start_list = df_start['i_count'].values.tolist()
    109.             end_list = df_end['i_count'].values.tolist()
    110.             for i,item in enumerate(start_list):
    111.                 signal_fiexed_target = pg.LinearRegionItem([item, end_list[i]],
    112.                                                            movable=False, brush=(
    113.                         self.color_line[0], self.color_line[1], self.color_line[2], 50))
    114.                 self.pw.addItem(signal_fiexed_target)
    115.             pass
    116.  
    117.         candle_fixed_target = CandlestickItem(candle_data)
    118.         self.main_fixed_target_list.append(candle_fixed_target)
    119.         self.pw.addItem(candle_fixed_target)
    120.  
    121.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    122.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    123.         self.label = pg.TextItem()
    124.  
    125.         self.pw.addItem(self.vLine, ignoreBounds=True)
    126.         self.pw.addItem(self.hLine, ignoreBounds=True)
    127.         self.pw.addItem(self.label, ignoreBounds=True)
    128.  
    129.         self.vb = self.pw.getViewBox()
    130.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    131.         self.pw.enableAutoRange()
    132.         pass
    133.  
    134.     def mouseMoved(self, evt):
    135.         pos = evt[0]
    136.         if self.pw.sceneBoundingRect().contains(pos):
    137.             mousePoint = self.vb.mapSceneToView(pos)
    138.             index = int(mousePoint.x())
    139.             if index >= 0 and index < len(self.current_whole_data):
    140.                 target_data = self.current_whole_data[index]
    141.                 html_str = ''
    142.                 for i, item in enumerate(self.whole_header):
    143.                     html_str += f"
      {item}:{target_data[i]}"
    144.                 self.label.setHtml(html_str)
    145.                 self.label.setPos(mousePoint.x(), mousePoint.y())
    146.             self.vLine.setPos(mousePoint.x())
    147.             self.hLine.setPos(mousePoint.y())
    148.         pass
    149.  
    150.     def mouseClicked(self, evt):
    151.         pass
    152.  
    153.     def updateViews(self):
    154.         pass
    155.     pass

    控制面板，也是主程序控件

    1. class StrategeMainWidget(QtWidgets.QWidget):
    2.     signal_runcode = QtCore.pyqtSignal(object)
    3.     signal_time = QtCore.pyqtSignal(object)
    4.     def __init__(self):
    5.         super().__init__()
    6.         self.thread_run: Thread = None
    7.         self.thread_time: Thread = None
    8.  
    9.         self.init_data()
    10.         self.init_ui()
    11.         self.register_event()
    12.         pass
    13.     def init_data(self):
    14.         self.pre_output_dir = './'
    15.         self.results_output_dir = './k_recognition_output/'
    16.         self.please_select_str: str = '--请选择--'
    17.         self.k_strategy_path: str = '../k_recognition_strategy/'
    18.         self.stratege_run_start_time = None
    19.         self.stratege_start = False
    20.         self.current_stratege_py_str: str = ''
    21.         self.dailydata_path:str = ''
    22.         self.k_strategy_map: Dict[str,Any] = {
    23.             '表上升和下跌的K线形态':['大阳线(长阳线)','大阴线(长阴线)','小阳线和小阴线','十字星和十字线','长十字线','螺旋桨','一字线','T字线和倒T字线','锤头线和吊颈线(绞刑线)','倒锤头线和射击之星(流星、扫帚星)','揉搓线','尽头线','镊子线','穿头破脚','身怀六甲和十字胎','下跌三连阴'],
    24.             '表见底和上升的K线形态':['向下加速度线','早晨(希望)之星和早晨(希望)十字星','好友反攻','曙光初现','旭日东升','平底','塔形底','圆底','低位并排阳线','低档五阳线','连续跳空三阴线','红三兵(三个白色武士)','冉冉上升','稳步上涨','徐缓上升','上升抵抗','上升(向上)弧形线','下探上涨','上涨两颗星','跳空上扬(升势鹤鸦缺口)','高位并排阳线(升势恋人肩并肩缺口)','跳空下跌三颗星','上升三部曲(升势三鸦)','多方尖兵(仙人指路)','两阳夹一阴(两红夹一黑)','上档盘旋形'],
    25.             '表见顶和下跌的K线形态':['向上加速度线','黄昏之星和黄昏十字星','淡友反攻','乌云盖顶','倾盆大雨','平顶(镊子顶)','塔形顶','圆顶','双飞乌鸦','高档五阴线','连续跳空三阳线','黑三兵','三只乌鸦(暴跌三杰)','绵绵阴跌','下跌不止','徐缓下降','下跌抵抗','下降弧形线','高开出逃','下跌三颗星','倒三阳','升势受阻','升势停顿','下降三部曲(降势三鹤)','空方尖兵','两阴夹一阳(两黑夹一红)','跛脚阳线','低档盘旋形','下降覆盖线']
    26.         }
    27.         self.strategyname_code_map:Dict[str,str] = {
    28.             '大阳线(长阳线)':'',
    29.             '大阴线(长阴线)':'',
    30.             '小阳线和小阴线':'',
    31.             '十字星和十字线':'',
    32.             '长十字线':'',
    33.             '螺旋桨':'',
    34.             '一字线':'',
    35.             'T字线和倒T字线':'',
    36.             '锤头线和吊颈线(绞刑线)':'',
    37.             '倒锤头线和射击之星(流星、扫帚星)':'',
    38.             '揉搓线':'',
    39.             '尽头线':'',
    40.             '镊子线':'',
    41.             '穿头破脚':'',
    42.             '身怀六甲和十字胎':'',
    43.             '下跌三连阴':'',
    44.             '向下加速度线':'',
    45.             '早晨(希望)之星和早晨(希望)十字星':'',
    46.             '好友反攻':'',
    47.             '曙光初现':'',
    48.             '旭日东升':'',
    49.             '平底':'',
    50.             '塔形底':'',
    51.             '圆底':'',
    52.             '低位并排阳线':'',
    53.             '低档五阳线':'',
    54.             '连续跳空三阴线':'',
    55.             '红三兵(三个白色武士)':'',
    56.             '冉冉上升':'',
    57.             '稳步上涨':'',
    58.             '徐缓上升':'',
    59.             '上升抵抗':'',
    60.             '上升(向上)弧形线':'',
    61.             '下探上涨':'',
    62.             '上涨两颗星':'',
    63.             '跳空上扬(升势鹤鸦缺口)':'',
    64.             '高位并排阳线(升势恋人肩并肩缺口)':'',
    65.             '跳空下跌三颗星':'',
    66.             '上升三部曲(升势三鸦)':'',
    67.             '多方尖兵(仙人指路)':'',
    68.             '两阳夹一阴(两红夹一黑)':'',
    69.             '上档盘旋形':'',
    70.             '向上加速度线':'',
    71.             '黄昏之星和黄昏十字星':'',
    72.             '淡友反攻':'',
    73.             '乌云盖顶':'',
    74.             '倾盆大雨':'',
    75.             '平顶(镊子顶)':'',
    76.             '塔形顶':'',
    77.             '圆顶':'',
    78.             '双飞乌鸦':'',
    79.             '高档五阴线':'',
    80.             '连续跳空三阳线':'',
    81.             '黑三兵':'',
    82.             '三只乌鸦(暴跌三杰)':'',
    83.             '绵绵阴跌':'',
    84.             '下跌不止':'',
    85.             '徐缓下降':'',
    86.             '下跌抵抗':'',
    87.             '下降弧形线':'',
    88.             '高开出逃':'',
    89.             '下跌三颗星':'',
    90.             '倒三阳':'',
    91.             '升势受阻':'',
    92.             '升势停顿':'',
    93.             '下降三部曲(降势三鹤)':'',
    94.             '空方尖兵':'',
    95.             '两阴夹一阳(两黑夹一红)':'',
    96.             '跛脚阳线':'',
    97.             '低档盘旋形':'',
    98.             '下降覆盖线':'',
    99.         }
    100.         self.table_header: List = ['日期','信号','备注']
    101.         pass
    102.     def init_ui(self):
    103.         self.setWindowTitle('股票K线形态识别工具')
    104.         tip_0 = QtWidgets.QLabel('股票日数据文件:')
    105.         self.dailydata_filepath_lineedit = QtWidgets.QLineEdit()
    106.         self.dailydata_filepath_lineedit.setReadOnly(True)
    107.         dailydata_choice_btn = QtWidgets.QPushButton('选择文件')
    108.         dailydata_choice_btn.clicked.connect(self.dailydata_choice_btn_clicked)
    109.  
    110.         layout_top = QtWidgets.QHBoxLayout()
    111.         layout_top.addWidget(tip_0)
    112.         layout_top.addWidget(self.dailydata_filepath_lineedit)
    113.         layout_top.addWidget(dailydata_choice_btn)
    114.         layout_top.addStretch(1)
    115.  
    116.         tip_1 = QtWidgets.QLabel('类别:')
    117.         self.type_combox = QtWidgets.QComboBox()
    118.         self.type_combox.addItem(self.please_select_str)
    119.         self.type_combox.addItems(list(self.k_strategy_map.keys()))
    120.         self.type_combox.currentIndexChanged.connect(self.type_combox_currentIndexChanged)
    121.  
    122.         tip_2 = QtWidgets.QLabel('形态:')
    123.         self.shape_combox = QtWidgets.QComboBox()
    124.         self.shape_combox.addItem(self.please_select_str)
    125.         self.shape_combox.currentIndexChanged.connect(self.shape_combox_currentIndexChanged)
    126.  
    127.         self.run_btn = QtWidgets.QPushButton('运行')
    128.         self.run_btn.clicked.connect(self.run_btn_clicked)
    129.  
    130.         self.time_label = QtWidgets.QLabel('')
    131.  
    132.         layout_combox = QtWidgets.QFormLayout()
    133.         layout_combox.addRow(tip_1,self.type_combox)
    134.         layout_combox.addRow(tip_2,self.shape_combox)
    135.         layout_combox.addRow(self.run_btn,self.time_label)
    136.  
    137.         self.code_textedit = QtWidgets.QTextEdit()
    138.         self.code_textedit.setReadOnly(True)
    139.  
    140.         self.results_table = PageTableWidget()
    141.         self.results_table.set_table_init_data({'headers':self.table_header})
    142.  
    143.         layout_left = QtWidgets.QVBoxLayout()
    144.         layout_left.addLayout(layout_combox)
    145.         layout_left.addWidget(self.code_textedit)
    146.         layout_left.addWidget(self.results_table)
    147.  
    148.         self.line_widget = PyQtGraphKWidget()
    149.  
    150.         layout_down = QtWidgets.QHBoxLayout()
    151.         layout_down.addLayout(layout_left,1)
    152.         layout_down.addWidget(self.line_widget,2)
    153.  
    154.         layout = QtWidgets.QVBoxLayout()
    155.         layout.addLayout(layout_top)
    156.         layout.addLayout(layout_down)
    157.         self.setLayout(layout)
    158.         pass
    159.     def register_event(self):
    160.         self.signal_runcode.connect(self.thread_run_excuted)
    161.         self.signal_time.connect(self.thread_time_excuted)
    162.         pass
    163.     def dailydata_choice_btn_clicked(self):
    164.         path, _ = QtWidgets.QFileDialog.getOpenFileName(
    165.             self,
    166.             '打开股票日数据所在文件',
    167.             self.pre_output_dir
    168.         )
    169.         if not path:
    170.             return
    171.         self.dailydata_filepath_lineedit.setText(path)
    172.         pass
    173.     def type_combox_currentIndexChanged(self,cur_i:int):
    174.         cur_txt = self.type_combox.currentText()
    175.         if not cur_txt or cur_txt == self.please_select_str:
    176.             return
    177.         shape_name_list = self.k_strategy_map[cur_txt]
    178.         self.shape_combox.clear()
    179.         self.shape_combox.addItem(self.please_select_str)
    180.         self.shape_combox.addItems(shape_name_list)
    181.         pass
    182.     def shape_combox_currentIndexChanged(self,cur_i:int):
    183.         cur_txt = self.shape_combox.currentText()
    184.         if not cur_txt or cur_txt == self.please_select_str:
    185.             return
    186.         cur_strategy_name = self.strategyname_code_map[cur_txt]
    187.         if len(cur_strategy_name)<=0:
    188.             QtWidgets.QMessageBox.information(
    189.                 self,
    190.                 '提示',
    191.                 '该K线形态的策略还没上线',
    192.                 QtWidgets.QMessageBox.Yes
    193.             )
    194.             return
    195.         strategy_file_path = self.k_strategy_path + cur_strategy_name
    196.         with open(strategy_file_path,'r',encoding='utf-8') as fr:
    197.             py_str = fr.read()
    198.         if len(py_str)<=20:
    199.             QtWidgets.QMessageBox.information(
    200.                 self,
    201.                 '提示',
    202.                 '策略文件代码为空',
    203.                 QtWidgets.QMessageBox.Yes
    204.             )
    205.             return
    206.         self.code_textedit.setPlainText(py_str)
    207.         pass
    208.     def run_btn_clicked(self):
    209.         '''运行按钮'''
    210.         # 检查股票日数据文件夹
    211.         dailydata_filepath = self.dailydata_filepath_lineedit.text()
    212.         dailydata_filepath = dailydata_filepath.strip()
    213.         if len(dailydata_filepath)<=0:
    214.             QtWidgets.QMessageBox.information(
    215.                 self,
    216.                 '提示',
    217.                 '请选择股票日数据文件',
    218.                 QtWidgets.QMessageBox.Yes
    219.             )
    220.             return
    221.  
    222.         py_str = self.code_textedit.toPlainText()
    223.         if len(py_str)<20:
    224.             QtWidgets.QMessageBox.information(
    225.                 self,
    226.                 '提示',
    227.                 '请选择要执行的策略',
    228.                 QtWidgets.QMessageBox.Yes
    229.             )
    230.             return
    231.         self.current_stratege_py_str = py_str
    232.  
    233.         self.run_btn.setDisabled(True)
    234.         self.shape_combox.setDisabled(True)
    235.         self.stratege_run_start_time = datetime.now()
    236.         self.stratege_start = True
    237.  
    238.         if self.thread_run:
    239.             QtWidgets.QMessageBox.information(
    240.                 self,
    241.                 '提示',
    242.                 '有策略正在运行',
    243.                 QtWidgets.QMessageBox.Yes
    244.             )
    245.             return
    246.         pre_data = {
    247.             'py_str':py_str,
    248.             'dailydata_filepath':dailydata_filepath
    249.         }
    250.         self.thread_run = Thread(
    251.             target=self.running_run_thread,
    252.             args=(pre_data,)
    253.         )
    254.         self.thread_run.start()
    255.         self.thread_time = Thread(
    256.             target=self.running_time_thread
    257.         )
    258.         self.thread_time.start()
    259.         pass
    260.     def running_run_thread(self,data:Dict[str,Any]):
    261.         '''执行代码线程'''
    262.         py_str = data['py_str']
    263.         dailydata_filepath = data['dailydata_filepath']
    264.  
    265.         namespace = {}
    266.         fun_stragegy = compile(py_str,'','exec')
    267.         exec(fun_stragegy,namespace)
    268.         ret = namespace['excute_strategy'](dailydata_filepath)
    269.         self.signal_runcode.emit(ret)
    270.         pass
    271.     def thread_run_excuted(self,data:Dict):
    272.         '''策略代码执行返回结果'''
    273.         self.run_btn.setDisabled(False)
    274.         self.shape_combox.setDisabled(False)
    275.  
    276.         # 保存结果文件
    277.         df = data['whole_df']
    278.         df00 = df.loc[df['signal']==1].copy()
    279.         table_data = df00.loc[:,['tradeDate','signal','signal_name']].values.tolist()
    280.         self.results_table.set_table_full_data(table_data)
    281.  
    282.         self.line_widget.set_data(data)
    283.  
    284.         self.thread_run = None
    285.         self.thread_time = None
    286.         self.stratege_start = False
    287.  
    288.         QtWidgets.QMessageBox.information(
    289.             self,
    290.             '提示',
    291.             '当前策略运行完毕',
    292.             QtWidgets.QMessageBox.Yes
    293.         )
    294.         pass
    295.     def running_time_thread(self):
    296.         '''计时线程'''
    297.         while self.stratege_start:
    298.             now = datetime.now()
    299.             interval_time = (now-self.stratege_run_start_time).seconds
    300.             res_map = {'res':interval_time}
    301.             self.signal_time.emit(res_map)
    302.             time.sleep(1)
    303.         pass
    304.     def thread_time_excuted(self,data:Dict):
    305.         '''计时返回结果'''
    306.         res = data['res']
    307.         self.time_label.setText(f"{res}s")
    308.         pass
    309.  
    310.     def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
    311.         if self.thread_time:
    312.             self.thread_time = None
    313.         if self.thread_run:
    314.             self.thread_run = None
    315.         self.close()

    工具使用

    1. if __name__ == '__main__':
    2.     QtCore.QCoreApplication.setAttribute(QtCore.Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    3.     app = QtWidgets.QApplication(sys.argv)
    4.     t_win = StrategeMainWidget()
    5.     t_win.showMaximized()
    6.     app.exec()
    7.     pass

    1. 在入口代码对应py文件的上一级目录下创建 k_recognition_strategy 文件夹，后续K线形态判别的策略代码都保存在这个文件夹下

     

    数据

     链接：https://pan.baidu.com/s/1acghH4GQyu_OaOUmEjPdHw 
    提取码：6toe

     

  • 相关阅读:
    html5新增_webStorage
    我眼中的大数据（一）
    C#开发的OpenRA游戏之系统参数选项按钮
    ubuntu 开启笔记本摄像头并修复画面颠倒问题
    数据结构 | 堆的向上调整和向下调整算法【奇妙的堆排序】
    haskell 基本布局和组成元素
    想要高效运行SolidWorks,云上设计了解一下
    后台任务 window.requestIdleCallback 方法的使用
    代码随想录 动态规划Ⅳ
    Hadoop学习记录1
  • 原文地址：https://blog.csdn.net/m0_37967652/article/details/127723113