• PyQt5_股票K线图相似度工具

    如果对某一股票K线图了解比较深刻，可以通过相似度找到与该股票相似的K线图，那么对这些相似的K线图，可以借鉴已经深刻剖析过的股票的方法，已达到事半功倍的效果。

    K线图相似度的计算方法使用的是该博文中的方法。

    目录

    效果

    计算K线图相似度策略代码

    工具代码

    工具使用

    数据

    效果

    计算K线图相似度策略代码

    前置说明：

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

    2. 三个参数，参数1为股票代码股票名键值对；参数2为股票日数据对应目录；参数3为指数（也可以是个股）日数据文件路径

    3. 策略代码写完后，保存为.py文件，将该py文件保存在一个目录（自定义）下，运行工具后，选择这个文件夹，程序会加载该目录下的所有py文件

    1. def excute_strategy(base_data,data_dir,index_data_dir):
    2.     '''
    3.     股票K线图相似度
    4.     计算 2021-06-01 到 2022-01-01 之间的日K线
    5.     :param base_data: 股票编码与股票名键值对
    6.     :param data_dir: 股票日数据文件夹目录
    7.     :param index_data_dir: 指数文件路径
    8.     :return:
    9.     '''
    10.     import pandas as pd
    11.     import numpy as np
    12.     import os
    13.  
    14.     start_date_str = '2021-06-01'
    15.     end_date_str = '2022-01-01'
    16.     index_df = pd.read_csv(index_data_dir,encoding='utf-8')
    17.     index_df['o_date'] = index_df['tradeDate']
    18.     index_df['o_date'] = pd.to_datetime(index_df['o_date'])
    19.     index_df = index_df.loc[(index_df['o_date']>=start_date_str) & (index_df['o_date']<=end_date_str)].copy()
    20.     index_df = index_df.loc[:,['o_date','openIndex','closeIndex','highestIndex','lowestIndex']].copy()
    21.  
    22.     dailydata_file_list = os.listdir(data_dir)
    23.  
    24.     res_list = []
    25.     for item in dailydata_file_list:
    26.         item_arr = item.split('.')
    27.         ticker = item_arr[0]
    28.         secName = base_data[ticker]
    29.         file_path = data_dir + item
    30.         df = pd.read_csv(file_path,encoding='utf-8')
    31.         # 删除停牌的数据
    32.         df = df.loc[df['openPrice'] > 0].copy()
    33.         df['o_date'] = df['tradeDate']
    34.         df['o_date'] = pd.to_datetime(df['o_date'])
    35.         df = df.loc[(df['o_date'] >= start_date_str) & (df['o_date']<=end_date_str)].copy()
    36.         # 保存未复权收盘价数据
    37.         df['close'] = df['closePrice']
    38.         # 计算前复权数据
    39.         df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    40.         df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    41.         df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    42.         df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    43.  
    44.         if len(df)<=0:
    45.             continue
    46.  
    47.         df_merge = pd.merge(index_df,df,on='o_date',how='inner')
    48.         corropen = np.corrcoef(df_merge['openIndex'],df_merge['openPrice'])[0][1]
    49.         corrclose = np.corrcoef(df_merge['closeIndex'],df_merge['closePrice'])[0][1]
    50.         corrhigh = np.corrcoef(df_merge['highestIndex'],df_merge['highestPrice'])[0][1]
    51.         corrlow = np.corrcoef(df_merge['lowestIndex'],df_merge['lowestPrice'])[0][1]
    52.  
    53.         similar_val = (corropen + corrclose + corrhigh + corrlow)/4
    54.         # similar_val = corrlow
    55.  
    56.         res_list.append({
    57.             'ticker':ticker,
    58.             'secName':secName,
    59.             'similar_val':similar_val
    60.         })
    61.         pass
    62.     df00 = pd.DataFrame(res_list)
    63.     df_similar = df00['similar_val'].value_counts(bins=10,sort=False)
    64.     similar_index_list00 = df_similar.index.tolist()
    65.     similar_bar_y = df_similar.values.tolist()
    66.     similar_bar_xTick = []
    67.     similar_bar_x = []
    68.     for i,item in enumerate(similar_index_list00):
    69.         similar_bar_x.append(i)
    70.         similar_bar_xTick.append((i,str(item)))
    71.         pass
    72.  
    73.     similar_bar_data = {
    74.         'title_str':'相似度分布',
    75.         'x':similar_bar_x,
    76.         'y':similar_bar_y,
    77.         'xTick':similar_bar_xTick
    78.     }
    79.  
    80.     # 指数文件名
    81.     index_file_name = os.path.basename(index_data_dir)
    82.  
    83.     results_data = {
    84.         'df':df00,
    85.         'start_date_str':start_date_str,
    86.         'end_date_str':end_date_str,
    87.         'index_file_name':index_file_name,
    88.         'bar_data':similar_bar_data
    89.     }
    90.     return results_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()

     K线图控件

    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('d'))
    13.         w = (self.data[1][0] - self.data[0][0]) / 3.
    14.         for (t, open, close, min, max) in self.data:
    15.             p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))
    16.             if open < close:
    17.                 p.setBrush(pg.mkBrush('r'))
    18.             else:
    19.                 p.setBrush(pg.mkBrush('g'))
    20.             p.drawRect(QtCore.QRectF(t-w, open, w * 2, close - open))
    21.         p.end()
    22.  
    23.     def paint(self, p, *args):
    24.         p.drawPicture(0, 0, self.picture)
    25.  
    26.     def boundingRect(self):
    27.         ## boundingRect _must_ indicate the entire area that will be drawn on
    28.         ## or else we will get artifacts and possibly crashing.
    29.         ## (in this case, QPicture does all the work of computing the bouning rect for us)
    30.         return QtCore.QRectF(self.picture.boundingRect())

    条形图控件

    1. class PyQtGraphHistWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.     def init_data(self):
    7.         self.color_bar = ()
    8.         pass
    9.     def init_ui(self):
    10.         self.title_label = QtWidgets.QLabel('均线排列状态')
    11.         self.title_label.setAlignment(Qt.AlignCenter)
    12.         xax = RotateAxisItem(orientation='bottom')
    13.         xax.setHeight(h=50)
    14.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    15.         self.pw.setMouseEnabled(x=True, y=False)
    16.         # self.pw.enableAutoRange(x=False,y=True)
    17.         self.pw.setAutoVisible(x=False, y=True)
    18.         layout = QtWidgets.QVBoxLayout()
    19.         layout.addWidget(self.title_label)
    20.         layout.addWidget(self.pw)
    21.         self.setLayout(layout)
    22.         pass
    23.     def set_data(self,data:Dict[str,Any]):
    24.         title_str = data['title_str']
    25.         x = data['x']
    26.         y = data['y']
    27.         xTick = [data['xTick']]
    28.  
    29.         self.y_datas = y
    30.         self.x_data = xTick
    31.         self.x_Tick = data['xTick']
    32.  
    33.         self.title_label.setText(title_str)
    34.         xax = self.pw.getAxis('bottom')
    35.         xax.setTicks(xTick)
    36.  
    37.         barItem = pg.BarGraphItem(x=x,height=y,width=0.8,brush=(107,200,224))
    38.         self.pw.addItem(barItem)
    39.  
    40.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    41.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    42.         self.label = pg.TextItem()
    43.  
    44.         self.pw.addItem(self.vLine, ignoreBounds=True)
    45.         self.pw.addItem(self.hLine, ignoreBounds=True)
    46.         self.pw.addItem(self.label, ignoreBounds=True)
    47.         self.vb = self.pw.getViewBox()
    48.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    49.         # 显示整条折线图
    50.         self.pw.enableAutoRange()
    51.         pass
    52.     def mouseMoved(self,evt):
    53.         pos = evt[0]
    54.         if self.pw.sceneBoundingRect().contains(pos):
    55.             mousePoint = self.vb.mapSceneToView(pos)
    56.             index = int(mousePoint.x())
    57.             if index >= 0 and index < len(self.y_datas):
    58.                 x_str = str(self.x_Tick[index][1])
    59.  
    60.                 y_str_html = ''
    61.                 y_str = str(self.y_datas[index])
    62.                 y_str_html += ' ' + y_str
    63.                 html_str = '
       ' + x_str +'
       '      +y_str_html+ '
      '
    64.                 self.label.setHtml(html_str)
    65.                 self.label.setPos(mousePoint.x(), mousePoint.y())
    66.             self.vLine.setPos(mousePoint.x())
    67.             self.hLine.setPos(mousePoint.y())
    68.         pass
    69.     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.         pass
    18.     def init_ui(self):
    19.         self.whole_duration_label = QtWidgets.QLabel('左边界~右边界')
    20.  
    21.         self.title_label = QtWidgets.QLabel('执行过程查看')
    22.         self.title_label.setAlignment(Qt.AlignCenter)
    23.         self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')
    24.  
    25.         xax = RotateAxisItem(orientation='bottom')
    26.         xax.setHeight(h=80)
    27.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    28.         self.pw.setMouseEnabled(x=True, y=True)
    29.         # self.pw.enableAutoRange(x=False,y=True)
    30.         self.pw.setAutoVisible(x=False, y=True)
    31.  
    32.         layout_right = QtWidgets.QVBoxLayout()
    33.         layout_right.addWidget(self.title_label)
    34.         layout_right.addWidget(self.whole_duration_label)
    35.         layout_right.addWidget(self.pw)
    36.         self.setLayout(layout_right)
    37.         pass
    38.  
    39.     def set_data(self, data: Dict[str, Any]):
    40.         title_str = data['title_str']
    41.         whole_header = data['whole_header']
    42.         whole_df = data['whole_df']
    43.         whole_pd_header = data['whole_pd_header']
    44.  
    45.         self.whole_header = whole_header
    46.         self.whole_df = whole_df
    47.         self.whole_pd_header = whole_pd_header
    48.  
    49.         self.title_label.setText(title_str)
    50.         self.whole_duration_label.setText(f"{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")
    51.  
    52.         self.current_whole_df = self.whole_df.copy()
    53.         self.caculate_and_show_data()
    54.         pass
    55.  
    56.     def caculate_and_show_data(self):
    57.         df = self.current_whole_df.copy()
    58.         df.reset_index(inplace=True)
    59.         df['i_count'] = [i for i in range(len(df))]
    60.         tradeDate_list = df['tradeDate'].values.tolist()
    61.         x = range(len(df))
    62.         xTick_show = []
    63.         x_dur = math.ceil(len(df) / 20)
    64.         for i in range(0, len(df), x_dur):
    65.             xTick_show.append((i, tradeDate_list[i]))
    66.         if len(df) % 20 != 0:
    67.             xTick_show.append((len(df) - 1, tradeDate_list[-1]))
    68.         candle_data = []
    69.         for i, row in df.iterrows():
    70.             candle_data.append(
    71.                 (row['i_count'], row['openPrice'], row['closePrice'], row['lowestPrice'], row['highestPrice']))
    72.  
    73.         self.current_whole_data = df.loc[:, self.whole_pd_header].values.tolist()
    74.         # 开始配置显示的内容
    75.         self.pw.clear()
    76.  
    77.         xax = self.pw.getAxis('bottom')
    78.         xax.setTicks([xTick_show])
    79.  
    80.         candle_fixed_target = CandlestickItem(candle_data)
    81.         self.main_fixed_target_list.append(candle_fixed_target)
    82.         self.pw.addItem(candle_fixed_target)
    83.  
    84.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    85.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    86.         self.label = pg.TextItem()
    87.  
    88.         self.pw.addItem(self.vLine, ignoreBounds=True)
    89.         self.pw.addItem(self.hLine, ignoreBounds=True)
    90.         self.pw.addItem(self.label, ignoreBounds=True)
    91.  
    92.         self.vb = self.pw.getViewBox()
    93.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    94.         self.pw.enableAutoRange()
    95.         pass
    96.  
    97.     def mouseMoved(self, evt):
    98.         pos = evt[0]
    99.         if self.pw.sceneBoundingRect().contains(pos):
    100.             mousePoint = self.vb.mapSceneToView(pos)
    101.             index = int(mousePoint.x())
    102.             if index >= 0 and index < len(self.current_whole_data):
    103.                 target_data = self.current_whole_data[index]
    104.                 html_str = ''
    105.                 for i, item in enumerate(self.whole_header):
    106.                     html_str += f"
      {item}:{target_data[i]}"
    107.                 self.label.setHtml(html_str)
    108.                 self.label.setPos(mousePoint.x(), mousePoint.y())
    109.             self.vLine.setPos(mousePoint.x())
    110.             self.hLine.setPos(mousePoint.y())
    111.         pass
    112.  
    113.     def mouseClicked(self, evt):
    114.         pass
    115.  
    116.     def updateViews(self):
    117.         pass
    118.     pass

    分页表格控件

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

     执行结果显示控件

    1. class PyQtGraphRunningWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.         pass
    7.     def init_data(self):
    8.         self.pre_output_dir = './'
    9.         self.results_output_dir = './results_output/'
    10.         self.json_file_name_list = []
    11.         self.dailydata_path: str = ''
    12.         self.indexdata_path: str = ''
    13.         self.table_header = ['股票编码','股票简称','相似度']
    14.         self.please_select_str = '---请选择---'
    15.         self.num_sort_map = {
    16.             '升序':True,
    17.             '降序':False
    18.         }
    19.         self.table_pd_header: List[str] = ['ticker','secName','similar_val']
    20.  
    21.         self.table_df = None
    22.         self.current_table_df = None
    23.         pass
    24.     def init_ui(self):
    25.         tip_5 = QtWidgets.QLabel('指数日数据文件夹:')
    26.         self.indexdata_dir_lineedit = QtWidgets.QLineEdit()
    27.         self.indexdata_dir_lineedit.setReadOnly(True)
    28.         indexdata_choice_btn = QtWidgets.QPushButton('选择文件夹')
    29.         indexdata_choice_btn.clicked.connect(self.indexdata_choice_btn_clicked)
    30.  
    31.         tip_0 = QtWidgets.QLabel('股票日数据文件夹:')
    32.         self.dailydata_dir_lineedit = QtWidgets.QLineEdit()
    33.         self.dailydata_dir_lineedit.setReadOnly(True)
    34.         dailydata_choice_btn = QtWidgets.QPushButton('选择文件夹')
    35.         dailydata_choice_btn.clicked.connect(self.dailydata_choice_btn_clicked)
    36.  
    37.         tip_1 = QtWidgets.QLabel('Json结果文件选择:')
    38.         self.json_combox = QtWidgets.QComboBox()
    39.         self.json_combox.addItem(self.please_select_str)
    40.         self.json_combox.currentIndexChanged.connect(self.json_combox_currentIndexChanged)
    41.         refresh_json_btn = QtWidgets.QPushButton('刷新结果下拉表')
    42.         refresh_json_btn.clicked.connect(self.refresh_json_btn_clicked)
    43.  
    44.         layout_top = QtWidgets.QGridLayout()
    45.         layout_top.addWidget(tip_5, 0, 0, 1, 1)
    46.         layout_top.addWidget(self.indexdata_dir_lineedit, 0, 1, 1, 3)
    47.         layout_top.addWidget(indexdata_choice_btn, 0, 4, 1, 1)
    48.         layout_top.addWidget(tip_0,1,0,1,1)
    49.         layout_top.addWidget(self.dailydata_dir_lineedit,1,1,1,3)
    50.         layout_top.addWidget(dailydata_choice_btn,1,4,1,1)
    51.         layout_top.addWidget(tip_1,2,0,1,1)
    52.         layout_top.addWidget(self.json_combox,2,1,1,3)
    53.         layout_top.addWidget(refresh_json_btn,2,4,1,1)
    54.  
    55.         tip_3 = QtWidgets.QLabel('股票名模糊查询')
    56.         self.query_lineedit = QtWidgets.QLineEdit()
    57.         query_btn = QtWidgets.QPushButton('查询')
    58.         query_btn.clicked.connect(self.query_btn_clicked)
    59.         reset_btn = QtWidgets.QPushButton('重置')
    60.         reset_btn.clicked.connect(self.reset_btn_clicked)
    61.  
    62.         tip_4 = QtWidgets.QLabel('持续时间排序:')
    63.         self.num_combox = QtWidgets.QComboBox()
    64.         self.num_combox.addItem(self.please_select_str)
    65.         self.num_combox.addItems(list(self.num_sort_map.keys()))
    66.         self.num_combox.currentIndexChanged.connect(self.num_combox_currentIndexChanged)
    67.  
    68.         layout_query = QtWidgets.QGridLayout()
    69.         layout_query.addWidget(tip_3,0,0,1,1)
    70.         layout_query.addWidget(self.query_lineedit,0,1,1,3)
    71.         layout_query.addWidget(query_btn,0,4,1,1)
    72.         layout_query.addWidget(reset_btn,0,5,1,1)
    73.         layout_query.addWidget(tip_4,1,0,1,1)
    74.         layout_query.addWidget(self.num_combox,1,4,1,2)
    75.  
    76.         self.table = PageTableWidget()
    77.         self.table.set_table_init_data({'headers': self.table_header})
    78.         self.table.output_signal.connect(self.table_output_signal_emit)
    79.  
    80.         self.bar_widget = PyQtGraphHistWidget()
    81.  
    82.         layout_left_up = QtWidgets.QVBoxLayout()
    83.         layout_left_up.addLayout(layout_query)
    84.         layout_left_up.addWidget(self.table)
    85.  
    86.         layout_left = QtWidgets.QVBoxLayout()
    87.         layout_left.addLayout(layout_left_up,3)
    88.         layout_left.addWidget(self.bar_widget,1)
    89.  
    90.         self.index_k_widget = PyQtGraphKWidget()
    91.         self.other_k_widget = PyQtGraphKWidget()
    92.  
    93.         layout_right = QtWidgets.QVBoxLayout()
    94.         layout_right.addWidget(self.other_k_widget)
    95.         layout_right.addWidget(self.index_k_widget)
    96.  
    97.         layout_down = QtWidgets.QHBoxLayout()
    98.         layout_down.addLayout(layout_left,1)
    99.         layout_down.addSpacing(30)
    100.         layout_down.addLayout(layout_right,2)
    101.  
    102.         layout = QtWidgets.QVBoxLayout()
    103.         layout.addLayout(layout_top)
    104.         layout.addLayout(layout_down)
    105.         self.setLayout(layout)
    106.         pass
    107.     def set_json_data(self,data:Dict[str,Any]):
    108.         self.table_df = data['df']
    109.         self.start_date_str = data['start_date_str']
    110.         self.end_date_str = data['end_date_str']
    111.  
    112.         self.current_table_df = self.table_df.copy()
    113.         self.fill_table_data()
    114.  
    115.         bar_data = data['bar_data']
    116.         self.bar_widget.set_data(bar_data)
    117.         pass
    118.     def fill_table_data(self):
    119.         table_data = self.current_table_df.loc[:, self.table_pd_header].values.tolist()
    120.         self.table.set_table_full_data(table_data)
    121.         pass
    122.  
    123.     def table_output_signal_emit(self,data:List):
    124.         # ticker secName status_name status_num
    125.         dailydata_dir = self.dailydata_dir_lineedit.text()
    126.         dailydata_dir = dailydata_dir.strip()
    127.         if len(dailydata_dir)<=0:
    128.             QtWidgets.QMessageBox.information(
    129.                 self,
    130.                 '提示',
    131.                 '请选择股票日数据文件件',
    132.                 QtWidgets.QMessageBox.Yes
    133.             )
    134.             return
    135.         daily_file_path = dailydata_dir + '/' + data[0] + '.csv'
    136.         df = pd.read_csv(daily_file_path,encoding='utf-8')
    137.         # 删除停牌的数据
    138.         df = df.loc[df['openPrice'] > 0].copy()
    139.         df['o_date'] = df['tradeDate']
    140.         df['o_date'] = pd.to_datetime(df['o_date'])
    141.         df = df.loc[(df['o_date'] >= self.start_date_str) & (df['o_date']<=self.end_date_str)].copy()
    142.         # 保存未复权收盘价数据
    143.         df['close'] = df['closePrice']
    144.         # 计算前复权数据
    145.         df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    146.         df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    147.         df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    148.         df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    149.  
    150.         columns_list = ['日期','收盘价','开盘价','最高价','最低价']
    151.         columns_pd_list = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice']
    152.  
    153.         line_data = {
    154.             'title_str':data[1],
    155.             'whole_header':columns_list,
    156.             'whole_df':df,
    157.             'whole_pd_header':columns_pd_list
    158.         }
    159.         self.other_k_widget.set_data(line_data)
    160.         pass
    161.     def indexdata_choice_btn_clicked(self):
    162.         path = QtWidgets.QFileDialog.getExistingDirectory(
    163.             self,
    164.             '打开指数日数据所在文件夹',
    165.             self.pre_output_dir
    166.         )
    167.         if not path:
    168.             return
    169.         self.indexdata_path = path
    170.         self.indexdata_dir_lineedit.setText(path)
    171.         pass
    172.     def dailydata_choice_btn_clicked(self):
    173.         path = QtWidgets.QFileDialog.getExistingDirectory(
    174.             self,
    175.             '打开股票日数据所在文件夹',
    176.             self.pre_output_dir
    177.         )
    178.         if not path:
    179.             return
    180.         self.dailydata_path = path
    181.         self.dailydata_dir_lineedit.setText(path)
    182.         pass
    183.     def json_combox_currentIndexChanged(self,cur_i:int):
    184.         cur_txt = self.json_combox.currentText()
    185.         if not cur_txt or cur_txt == self.please_select_str:
    186.             return
    187.         current_json_file_path = self.results_output_dir + cur_txt
    188.         with open(current_json_file_path,'r',encoding='utf-8') as fr:
    189.             obj_json = json.load(fr)
    190.         index_file_name = obj_json['index_file_name']
    191.         start_date_str = obj_json['start_date_str']
    192.         end_date_str = obj_json['end_date_str']
    193.         indexdata_dir = self.indexdata_dir_lineedit.text()
    194.         indexdata_dir = indexdata_dir.strip()
    195.         if len(indexdata_dir) <= 0:
    196.             QtWidgets.QMessageBox.information(
    197.                 self,
    198.                 '提示',
    199.                 '请先选择指数日数据所在文件夹',
    200.                 QtWidgets.QMessageBox.Yes
    201.             )
    202.             return
    203.         indexfile_path = indexdata_dir +'/'+ index_file_name
    204.         if not os.path.exists(indexfile_path):
    205.             QtWidgets.QMessageBox.information(
    206.                 self,
    207.                 '提示',
    208.                 '不存在，'+index_file_name,
    209.                 QtWidgets.QMessageBox.Yes
    210.             )
    211.             return
    212.  
    213.         # 显示指数K线图
    214.         df_index = pd.read_csv(indexfile_path,encoding='utf-8')
    215.         df_index['o_date'] = df_index['tradeDate']
    216.         df_index['o_date'] = pd.to_datetime(df_index['o_date'])
    217.         df_index = df_index.loc[(df_index['o_date']>=start_date_str) & (df_index['o_date']<=end_date_str)].copy()
    218.         df_index['openPrice'] = df_index['openIndex']
    219.         df_index['closePrice'] = df_index['closeIndex']
    220.         df_index['highestPrice'] = df_index['highestIndex']
    221.         df_index['lowestPrice'] = df_index['lowestIndex']
    222.         title_str = index_file_name.split('.')[0]
    223.         whole_header = ['日期','收','开','高','低']
    224.         whole_df = df_index
    225.         whole_pd_header = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice']
    226.  
    227.         index_line_data = {
    228.             'title_str':title_str,
    229.             'whole_header':whole_header,
    230.             'whole_df':whole_df,
    231.             'whole_pd_header':whole_pd_header
    232.         }
    233.         self.index_k_widget.set_data(index_line_data)
    234.  
    235.         df = pd.DataFrame(obj_json['df_json'])
    236.         obj_json['df'] = df
    237.  
    238.         self.set_json_data(obj_json)
    239.         pass
    240.     def query_btn_clicked(self):
    241.         query_str = self.query_lineedit.text()
    242.         query_str = query_str.strip()
    243.         if len(query_str)<=0:
    244.             QtWidgets.QMessageBox.information(
    245.                 self,
    246.                 '提示',
    247.                 '请输入要查询的内容',
    248.                 QtWidgets.QMessageBox.Yes
    249.             )
    250.             return
    251.         self.status_combox.setCurrentText(self.please_select_str)
    252.         df = self.table_df.copy()
    253.         self.current_table_df = df.loc[df['secName'].str.contains(query_str)].copy()
    254.         self.fill_table_data()
    255.         pass
    256.     def reset_btn_clicked(self):
    257.         self.query_lineedit.setText('')
    258.         self.status_combox.setCurrentText(self.please_select_str)
    259.         self.current_table_df = self.table_df.copy()
    260.         self.fill_table_data()
    261.         pass
    262.     def num_combox_currentIndexChanged(self,cur_i:int):
    263.         cur_txt = self.num_combox.currentText()
    264.         if cur_txt == self.please_select_str:
    265.             return
    266.         self.current_table_df.sort_values(by='similar_val',ascending=self.num_sort_map[cur_txt],inplace=True)
    267.         self.fill_table_data()
    268.         pass
    269.     def refresh_json_btn_clicked(self):
    270.         indexdata_dir = self.indexdata_dir_lineedit.text()
    271.         indexdata_dir = indexdata_dir.strip()
    272.         if len(indexdata_dir)<=0:
    273.             QtWidgets.QMessageBox.information(
    274.                 self,
    275.                 '提示',
    276.                 '请先选择指数日数据所在文件夹',
    277.                 QtWidgets.QMessageBox.Yes
    278.             )
    279.             return
    280.         file_list = os.listdir(self.results_output_dir)
    281.         json_file_list = []
    282.         for item in file_list:
    283.             if item.endswith('.json'):
    284.                 json_file_list.append(item)
    285.         self.json_file_name_list.extend(json_file_list)
    286.         json_file_set = set(self.json_file_name_list)
    287.         self.json_file_name_list = list(json_file_set)
    288.         self.json_combox.clear()
    289.         self.json_combox.addItem(self.please_select_str)
    290.         self.json_combox.addItems(self.json_file_name_list)
    291.         pass
    292.     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.running_graph_widget: QtWidgets.QWidget = None
    10.  
    11.         self.init_data()
    12.         self.init_ui()
    13.         self.register_event()
    14.         pass
    15.     def init_data(self):
    16.         self.pre_output_dir = './'
    17.         self.results_output_dir = './results_output/'
    18.         self.secID_name_file_name = 'secID_name.csv'
    19.         self.please_select_str: str = '--请选择--'
    20.         self.stratege_name_list: List = []
    21.         self.tip_msg_0: str = '1.选择策略所在文件夹；2.选择策略；3.点击运行。'
    22.         self.stratege_path: str = ''
    23.         self.stratege_run_start_time = None
    24.         self.stratege_start = False
    25.         self.current_stratege_py_str: str = ''
    26.         self.dailydata_path:str = ''
    27.         pass
    28.     def init_ui(self):
    29.         self.setWindowTitle('股票K线图相似度工具')
    30.         tip_3 = QtWidgets.QLabel('指数日数据文件:')
    31.         self.indexfilepath_lineedit = QtWidgets.QLineEdit()
    32.         self.indexfilepath_lineedit.setReadOnly(True)
    33.         indexfilepath_choice_btn = QtWidgets.QPushButton('选择文件')
    34.         indexfilepath_choice_btn.clicked.connect(self.indexfilepath_choice_btn_clicked)
    35.  
    36.         tip_2 = QtWidgets.QLabel('股票日数据文件夹:')
    37.         self.dailydata_dir_lineedit = QtWidgets.QLineEdit()
    38.         self.dailydata_dir_lineedit.setReadOnly(True)
    39.         dailydata_choice_btn = QtWidgets.QPushButton('选择文件夹')
    40.         dailydata_choice_btn.clicked.connect(self.dailydata_choice_btn_clicked)
    41.  
    42.         tip_0 = QtWidgets.QLabel('选择策略所在文件夹:')
    43.         self.stratege_dir_lineedit = QtWidgets.QLineEdit()
    44.         self.stratege_dir_lineedit.setReadOnly(True)
    45.         stratege_choice_btn = QtWidgets.QPushButton('选择文件夹')
    46.         stratege_choice_btn.clicked.connect(self.stratege_choice_btn_clicked)
    47.  
    48.         tip_1 = QtWidgets.QLabel('策略:')
    49.         self.stratege_combox = QtWidgets.QComboBox()
    50.         self.stratege_combox.addItem(self.please_select_str)
    51.         self.stratege_combox.currentIndexChanged.connect(self.stratege_combox_currentIndexChanged)
    52.  
    53.         self.run_btn = QtWidgets.QPushButton('运行')
    54.         self.run_btn.clicked.connect(self.run_btn_clicked)
    55.         self.force_stop_btn = QtWidgets.QPushButton('强制停止')
    56.         self.force_stop_btn.clicked.connect(self.force_stop_btn_clicked)
    57.  
    58.         layout_top_left = QtWidgets.QGridLayout()
    59.         layout_top_left.addWidget(tip_3, 0, 0, 1, 1)
    60.         layout_top_left.addWidget(self.indexfilepath_lineedit, 0, 1, 1, 3)
    61.         layout_top_left.addWidget(indexfilepath_choice_btn, 0, 4, 1, 1)
    62.         layout_top_left.addWidget(tip_2,1,0,1,1)
    63.         layout_top_left.addWidget(self.dailydata_dir_lineedit,1,1,1,3)
    64.         layout_top_left.addWidget(dailydata_choice_btn,1,4,1,1)
    65.         layout_top_left.addWidget(tip_0,2,0,1,1)
    66.         layout_top_left.addWidget(self.stratege_dir_lineedit,2,1,1,3)
    67.         layout_top_left.addWidget(stratege_choice_btn,2,4,1,1)
    68.         layout_top_left.addWidget(tip_1,3,0,1,1)
    69.         layout_top_left.addWidget(self.stratege_combox,3,1,1,2)
    70.         layout_top_left.addWidget(self.run_btn,3,3,1,1)
    71.         layout_top_left.addWidget(self.force_stop_btn,3,4,1,1)
    72.  
    73.         self.tip_msg_label = QtWidgets.QLabel()
    74.         self.tip_msg_label.setWordWrap(True)
    75.         self.tip_msg_label.setText(self.tip_msg_0)
    76.         results_output_look_btn = QtWidgets.QPushButton('结果查看')
    77.         results_output_look_btn.clicked.connect(self.results_output_look_btn_clicked)
    78.  
    79.         layout_top_right = QtWidgets.QHBoxLayout()
    80.         layout_top_right.addWidget(self.tip_msg_label)
    81.         layout_top_right.addWidget(results_output_look_btn)
    82.  
    83.         layout_top = QtWidgets.QHBoxLayout()
    84.         layout_top.addLayout(layout_top_left,3)
    85.         layout_top.addSpacing(30)
    86.         layout_top.addLayout(layout_top_right,1)
    87.  
    88.         self.code_textedit = QtWidgets.QTextEdit()
    89.         self.code_textedit.setReadOnly(True)
    90.  
    91.         layout = QtWidgets.QVBoxLayout()
    92.         layout.addLayout(layout_top)
    93.         layout.addWidget(self.code_textedit)
    94.         self.setLayout(layout)
    95.         pass
    96.     def register_event(self):
    97.         self.signal_runcode.connect(self.thread_run_excuted)
    98.         self.signal_time.connect(self.thread_time_excuted)
    99.         pass
    100.     def results_output_look_btn_clicked(self):
    101.         '''策略运行结果查看'''
    102.         if not self.running_graph_widget:
    103.             self.running_graph_widget = PyQtGraphRunningWidget()
    104.         self.running_graph_widget.showMaximized()
    105.         pass
    106.     def indexfilepath_choice_btn_clicked(self):
    107.         path, _ = QtWidgets.QFileDialog.getOpenFileName(
    108.             self,
    109.             '打开指数日数据文件',
    110.             self.pre_output_dir
    111.         )
    112.         if not path:
    113.             return
    114.         self.indexfilepath_lineedit.setText(path)
    115.         pass
    116.     def dailydata_choice_btn_clicked(self):
    117.         path = QtWidgets.QFileDialog.getExistingDirectory(
    118.             self,
    119.             '打开股票日数据所在文件夹',
    120.             self.pre_output_dir
    121.         )
    122.         if not path:
    123.             return
    124.         self.dailydata_path = path+'/'
    125.         self.dailydata_dir_lineedit.setText(path)
    126.         pass
    127.     def stratege_choice_btn_clicked(self):
    128.         '''选择策略所在文件夹'''
    129.         path = QtWidgets.QFileDialog.getExistingDirectory(
    130.             self,
    131.             '打开策略所在文件夹',
    132.             self.pre_output_dir
    133.         )
    134.         if not path:
    135.             return
    136.         self.stratege_path = path
    137.         self.stratege_dir_lineedit.setText(path)
    138.         file_list = os.listdir(path)
    139.         temp_file_list = set(self.stratege_name_list)
    140.         for item in file_list:
    141.             if item.endswith('.py'):
    142.                 temp_file_list.add(item)
    143.         self.stratege_name_list = list(temp_file_list)
    144.  
    145.         self.stratege_combox.clear()
    146.         self.stratege_combox.addItem(self.please_select_str)
    147.         self.stratege_combox.addItems(self.stratege_name_list)
    148.         pass
    149.     def stratege_combox_currentIndexChanged(self,cur_i:int):
    150.         cur_txt = self.stratege_combox.currentText()
    151.         if not cur_txt or cur_txt == self.please_select_str:
    152.             self.code_textedit.clear()
    153.             return
    154.         file_path = self.stratege_path + os.path.sep + cur_txt
    155.         with open(file_path,'r',encoding='utf-8') as fr:
    156.             code_txt = fr.read()
    157.         self.code_textedit.setPlainText(code_txt)
    158.         pass
    159.     def run_btn_clicked(self):
    160.         '''运行按钮'''
    161.         # 检查指数日数据文件
    162.         indexfilepath = self.indexfilepath_lineedit.text()
    163.         indexfilepath = indexfilepath.strip()
    164.         if len(indexfilepath)<=0:
    165.             QtWidgets.QMessageBox.information(
    166.                 self,
    167.                 '提示',
    168.                 '请选择指数日数据文件',
    169.                 QtWidgets.QMessageBox.Yes
    170.             )
    171.             return
    172.         # 检查股票日数据文件夹
    173.         dailydata_dir = self.dailydata_dir_lineedit.text()
    174.         dailydata_dir = dailydata_dir.strip()
    175.         if len(dailydata_dir)<=0:
    176.             QtWidgets.QMessageBox.information(
    177.                 self,
    178.                 '提示',
    179.                 '请选择股票日数据文件夹',
    180.                 QtWidgets.QMessageBox.Yes
    181.             )
    182.             return
    183.         dailydata_file_list = os.listdir(dailydata_dir)
    184.         if len(dailydata_file_list)<=0:
    185.             QtWidgets.QMessageBox.information(
    186.                 self,
    187.                 '提示',
    188.                 '股票日数据文件夹中没有文件',
    189.                 QtWidgets.QMessageBox.Yes
    190.             )
    191.             return
    192.         secID_name_file = self.results_output_dir + self.secID_name_file_name
    193.         if not os.path.exists(secID_name_file):
    194.             QtWidgets.QMessageBox.information(
    195.                 self,
    196.                 '提示',
    197.                 '股票码与股票名基础文件不存在',
    198.                 QtWidgets.QMessageBox.Yes
    199.             )
    200.             return
    201.  
    202.         py_str = self.code_textedit.toPlainText()
    203.         if len(py_str)<10:
    204.             QtWidgets.QMessageBox.information(
    205.                 self,
    206.                 '提示',
    207.                 '请选择要执行的策略',
    208.                 QtWidgets.QMessageBox.Yes
    209.             )
    210.             return
    211.         self.current_stratege_py_str = py_str
    212.         base_data = {}
    213.         base_df = pd.read_csv(secID_name_file,encoding='utf-8')
    214.         for i,row in base_df.iterrows():
    215.             secID = row['secID']
    216.             secID_arr = secID.split('.')
    217.             base_data[secID_arr[0]] = row['secShortName']
    218.             pass
    219.  
    220.         self.run_btn.setDisabled(True)
    221.         self.stratege_combox.setDisabled(True)
    222.         self.stratege_run_start_time = datetime.now()
    223.         self.stratege_start = True
    224.  
    225.         if self.thread_run:
    226.             QtWidgets.QMessageBox.information(
    227.                 self,
    228.                 '提示',
    229.                 '有策略正在运行',
    230.                 QtWidgets.QMessageBox.Yes
    231.             )
    232.             return
    233.         pre_data = {
    234.             'py_str':py_str,
    235.             'base_data':base_data,
    236.             'indexfilepath':indexfilepath
    237.         }
    238.         self.thread_run = Thread(
    239.             target=self.running_run_thread,
    240.             args=(pre_data,)
    241.         )
    242.         self.thread_run.start()
    243.         self.thread_time = Thread(
    244.             target=self.running_time_thread
    245.         )
    246.         self.thread_time.start()
    247.         pass
    248.     def force_stop_btn_clicked(self):
    249.         '''强制停止按钮'''
    250.         self.thread_run = None
    251.         self.thread_time = None
    252.  
    253.         self.run_btn.setDisabled(False)
    254.         self.stratege_combox.setDisabled(False)
    255.         self.stratege_start = False
    256.         pass
    257.     def running_run_thread(self,data:Dict[str,Any]):
    258.         '''执行代码线程'''
    259.         py_str = data['py_str']
    260.         base_data = data['base_data']
    261.         indexfilepath = data['indexfilepath']
    262.  
    263.         namespace = {}
    264.         fun_stragegy = compile(py_str,'','exec')
    265.         exec(fun_stragegy,namespace)
    266.         ret = namespace['excute_strategy'](base_data,self.dailydata_path,indexfilepath)
    267.         self.signal_runcode.emit(ret)
    268.         pass
    269.     def running_time_thread(self):
    270.         '''计时线程'''
    271.         while self.stratege_start:
    272.             now = datetime.now()
    273.             interval_time = (now-self.stratege_run_start_time).seconds
    274.             res_map = {'res':interval_time}
    275.             self.signal_time.emit(res_map)
    276.             time.sleep(1)
    277.         pass
    278.     def thread_run_excuted(self,data:Dict):
    279.         '''策略代码执行返回结果'''
    280.         self.run_btn.setDisabled(False)
    281.         self.stratege_combox.setDisabled(False)
    282.  
    283.         # 保存结果文件
    284.         now_datetime_str = datetime.now().strftime('%Y%m%d%H%M%S')
    285.         df = data['df']
    286.         df_json = df.to_dict(orient='records')
    287.         pre_save_data = {
    288.             'df_json':df_json,
    289.             'end_date_str':data['end_date_str'],
    290.             'start_date_str':data['start_date_str'],
    291.             'index_file_name':data['index_file_name'],
    292.             'bar_data':data['bar_data']
    293.         }
    294.         with open(self.results_output_dir + now_datetime_str + '.json','w',encoding='utf-8') as fw:
    295.             json.dump(pre_save_data,fw)
    296.         if not self.running_graph_widget:
    297.             self.running_graph_widget = PyQtGraphRunningWidget()
    298.         self.running_graph_widget.showMaximized()
    299.         self.thread_run = None
    300.         self.thread_time = None
    301.         self.stratege_start = False
    302.  
    303.         QtWidgets.QMessageBox.information(
    304.             self,
    305.             '提示',
    306.             '当前策略运行完毕',
    307.             QtWidgets.QMessageBox.Yes
    308.         )
    309.         pass
    310.     def thread_time_excuted(self,data:Dict):
    311.         '''计时返回结果'''
    312.         res = data['res']
    313.         self.tip_msg_label.setText(f"{res}s")
    314.         pass
    315.     def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
    316.         if self.thread_time:
    317.             self.thread_time.join()
    318.         if self.thread_run:
    319.             self.thread_run.join()
    320.         if self.running_graph_widget:
    321.             self.running_graph_widget.close()
    322.         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文件的同一目录下创建 results_output 目录

    2. 下载数据，将 secID_name.csv 文件放入 results_output 目录里

    3. 程序运行策略后，会把执行结果放入 results_output 目录中，以json文件存储

     

    数据

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

  • 相关阅读:
    go语言面试
    Java练习题-输出二维数组对角线元素和
    新建一个ARXML数据库文件
    网络安全总结
    二十三种设计模式全面解析-迭代器模式进阶篇：探索变体与扩展
    C++中的文件输入/输出
    java计算机毕业设计酒店预订管理系统源码+mysql数据库+系统+lw文档+部署
    HFSS笔记——求解器和求解分析
    自学WEB后端01-安装Express+Node.js框架完成Hello World!
    网络协议--ARP：地址解析协议
  • 原文地址：https://blog.csdn.net/m0_37967652/article/details/127659195