• PyQt5_pyqtgraph双均线组合工具

    目录

    双线组合概念

    双线组合交易原则

    交易原则翻译为代码逻辑

    策略代码实现

    运行工具代码

    执行

    验证策略思想

    优化策略

    数据

    双线组合概念

            双线组合又称两条均线组合，是由一条时间周期较短的均线和一条时间周期较长的均线组合而成的均线分析系统。

            其中时间周期较长的均线，主要用于预测和判断趋势方向。时间周期较短的均线，主要利用其与股价、长周期的均线之间的相互位置和关系，作为确定进出场的依据。为便于解说，时间周期较长的均线，我们称之为定性线；时间周期较短的均线，我们称之为定量线。

    PS:以上概念来自书籍《均线技术分析》by 邱立波

    双线组合交易原则

    1）买入和持仓原则。

    A. 股价向上突破定性线，定性线上行，买入。

    B. 定量线上穿定性线形成黄金交叉，买入

    C. 股价下跌，遇定性线上行支撑止跌回升，买入

    D. 定性线上行，股价在定性线上方向上突破定量线，买入

    E. 定量线下行，遇定性线上行支撑止跌后再度上升，买入。

    F. 股价、定量线、定性线多头排列，持股

    2）卖出和空仓原则

    A. 股价跌破定性线，定性线走平或已拐头下行，卖出

    B. 多头排列期间，股价跌破定量线，减仓

    C. 股价急速飙升，远离定性线，减仓

    D. 定性线下行，空仓

    PS:以上交易原则来自书籍《均线技术分析》by 邱立波

    交易原则翻译为代码逻辑

             一些概念诸如上行、飙升、下跌等描述要置换成代码逻辑，需要进行开发者做自定义的量化，这部分量化会影响信号发出的频次，先定义好一个默认量，后续策略优化有需要再行修改。

    本文策略自定义规则：

    描述代码逻辑规则（自定义）
    支撑和阻碍股价与均线价格之间的差值小于等于当前股价的2%
    股价下跌连续三日收盘价涨跌幅为负        
    均线下行连续三日均线涨跌幅为负
    股价急速飙升连续三日股价涨跌幅都大于等于5%
    远离股价与均线价格之间的差值大于等于当前股价的50%

     下面开始逐句翻译交易原则：

    1) 买入和持仓原则：
        A. 股价向上突破定性线，定性线上行，买入。
        翻译：收盘价涨跌幅为正 & 前一日收盘价小于定性价格 & 收盘价大于定性价格 & 定性线连续三日涨跌幅为正
        B. 定量线上穿定性线形成黄金交叉，买入。
        翻译：前一日定量价格小于前一日定性价格 & 当日定量价格大于定性价格
        C. 股价下跌，遇定性线上行支撑止跌回升，买入。
        翻译：连续三日收盘价涨跌幅为负 & 定性线连续三日涨跌幅为正 & 股价被定性线支撑 & 下一日收盘价正跌幅为正
        D. 定性线上行，股价在定性线上方向上突破定量线，买入。
        翻译：定性线连续三日涨跌幅为正 & 收盘价大于定性线价格 & （前一日收盘价小于前一日定量价格 & 当日收盘价大于定量价格）
        E. 定量线下行，遇定性线上行支撑止跌后再度上升，买入。
        翻译：连续三日定量线涨跌幅为负 & 连续三日定性线涨跌幅为正 & 定量线被定性线支撑 & 下一日定量线涨跌幅为正
        F. 股价、定量线、定性线多头排列，持股
        翻译：收盘价大于定量价格大于定性价格
        2） 卖出和空仓原则
        A. 股价跌破定性线，定性线走平或已拐头下行，卖出
        翻译：前一日收盘价大于定性价格 & 当日收盘价小于定性价格 & 定性价格涨跌幅小于等于0
        B. 多头排列期间，股价跌破定量线，减仓
        翻译：前一日收盘价大于定量价格大于定性价格 & 当日收盘价小于定量价格
        C. 股价急速飙升，远离定性线，减仓
        翻译：股价飙升 & 收盘价远离定性价格
        D. 定性线下行，空仓
        翻译：连续三日定性价格涨跌幅为负

    策略代码实现

    本文策略使用ma5作为定量线，ma20作为定性线 

    1. def excute_strategy():
    2.     '''
    3.     前置自定义规则：
    4.     A. 支撑和阻碍，定义为股价与均线价格之间的差值小于等于当前股价的2%
    5.     B. 股价下跌，连续三日收盘价涨跌幅为负
    6.     C. 均线下行，连续三日均线涨跌幅为负
    7.     D. 股价急速飙升，连续三日股价涨跌幅都大于等于5%
    8.     E. 远离，定义为股价与均线价格之间的差值大于等于当前股价的50%
    9.     1) 买入和持仓原则：
    10.     A. 股价向上突破定性线，定性线上行，买入。
    11.     翻译：收盘价涨跌幅为正 & 前一日收盘价小于定性价格 & 收盘价大于定性价格 & 定性线连续三日涨跌幅为正
    12.     B. 定量线上穿定性线形成黄金交叉，买入。
    13.     翻译：前一日定量价格小于前一日定性价格 & 当日定量价格大于定性价格
    14.     C. 股价下跌，遇定性线上行支撑止跌回升，买入。
    15.     翻译：连续三日收盘价涨跌幅为负 & 定性线连续三日涨跌幅为正 & 股价被定性线支撑 & 下一日收盘价正跌幅为正
    16.     D. 定性线上行，股价在定性线上方向上突破定量线，买入。
    17.     翻译：定性线连续三日涨跌幅为正 & 收盘价大于定性线价格 & （前一日收盘价小于前一日定量价格 & 当日收盘价大于定量价格）
    18.     E. 定量线下行，遇定性线上行支撑止跌后再度上升，买入。
    19.     翻译：连续三日定量线涨跌幅为负 & 连续三日定性线涨跌幅为正 & 定量线被定性线支撑 & 下一日定量线涨跌幅为正
    20.     F. 股价、定量线、定性线多头排列，持股
    21.     翻译：收盘价大于定量价格大于定性价格
    22.     2） 卖出和空仓原则
    23.     A. 股价跌破定性线，定性线走平或已拐头下行，卖出
    24.     翻译：前一日收盘价大于定性价格 & 当日收盘价小于定性价格 & 定性价格涨跌幅小于等于0
    25.     B. 多头排列期间，股价跌破定量线，减仓
    26.     翻译：前一日收盘价大于定量价格大于定性价格 & 当日收盘价小于定量价格
    27.     C. 股价急速飙升，远离定性线，减仓
    28.     翻译：股价飙升 & 收盘价远离定性价格
    29.     D. 定性线下行，空仓
    30.     翻译：连续三日定性价格涨跌幅为负
    31.     :return:
    32.     '''
    33.     import pandas as pd
    34.     import talib
    35.     df = pd.read_csv('E:/temp005/600660.csv',encoding='utf-8')
    36.     # 删除停牌的数据
    37.     df = df.loc[df['openPrice'] > 0].copy()
    38.     df['o_date'] = df['tradeDate']
    39.     df['o_date'] = pd.to_datetime(df['o_date'])
    40.     df = df.loc[df['o_date']>='2020-01-01'].copy()
    41.     # 保存未复权收盘价数据
    42.     df['close'] = df['closePrice']
    43.     # 计算前复权数据
    44.     df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    45.     df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    46.     df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    47.     df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    48.  
    49.     # 双均线组合
    50.     # 定性线：20日均线  定量线：5日均线
    51.     df['ma5'] = talib.MA(df['closePrice'],timeperiod=5)
    52.     df['ma20'] = talib.MA(df['closePrice'],timeperiod=20)
    53.     # 定性线涨跌幅
    54.     df['ma20_rate'] = df['ma20']-df['ma20'].shift(1)
    55.     # 定性线上行还是下行:连续三天涨跌幅为正，为上行，值为1；连续三天涨跌幅为负，为下行，值为-1
    56.     df['ma20_direction'] = 0
    57.     df.loc[(df['ma20_rate']>0) & (df['ma20_rate'].shift(1)>0) & (df['ma20_rate'].shift(2)>0),'ma20_direction'] = 1
    58.     df.loc[(df['ma20_rate']<0) & (df['ma20_rate'].shift(1)<0) & (df['ma20_rate'].shift(2)<0),'ma20_direction'] = -1
    59.     # 定量线涨跌幅
    60.     df['ma5_rate'] = df['ma5'] - df['ma5'].shift(1)
    61.     df['ma5_direction'] = 0
    62.     df.loc[(df['ma5_rate']>0) & (df['ma5_rate'].shift(1)>0) & (df['ma5_rate'].shift(2)>0),'ma5_direction'] = 1
    63.     df.loc[(df['ma5_rate']<0) & (df['ma5_rate'].shift(1)<0) & (df['ma5_rate'].shift(2)<0),'ma5_direction'] = -1
    64.     # 收盘价涨跌幅
    65.     df['close_rate'] = df['closePrice'] - df['closePrice'].shift(1)
    66.     df['closePrice_direction'] = 0
    67.     df.loc[(df['close_rate'] > 0) & (df['close_rate'].shift(1) > 0) & (df['close_rate'].shift(2) > 0), 'closePrice_direction'] = 1
    68.     df.loc[(df['close_rate'] < 0) & (df['close_rate'].shift(1) < 0) & (df['close_rate'].shift(2) < 0), 'closePrice_direction'] = -1
    69.     # 收盘价到定性线之间的距离
    70.     df['close_ma20_distance'] = (abs(df['closePrice']-df['ma20'])/df['closePrice'])*100
    71.     # 收盘价到定量线之间的距离
    72.     df['close_ma5_distance'] = (abs(df['closePrice']-df['ma5'])/df['closePrice'])*100
    73.     # 定量线到定性线之间的距离
    74.     df['ma5_ma20_distance'] = (abs(df['ma5']-df['ma20'])/df['ma5'])*100
    75.     # 定性线到定量线之间的距离
    76.     df['ma20_ma5_distance'] = (abs(df['ma5']-df['ma20'])/df['ma20'])*100
    77.     # 股价是否飙升
    78.     df['pre_fast_up'] = 0
    79.     df.loc[(df['close_rate']>0) & (df['close_rate']/df['closePrice']>=0.05),'pre_fast_up'] = 1
    80.     df['fast_up'] = 0
    81.     df.loc[(df['pre_fast_up']==1) & (df['pre_fast_up'].shift(1)==1) & (df['pre_fast_up'].shift(2)==1),'fast_up'] = 1
    82.  
    83.     # 1) 买入和持仓原则：
    84.     # 买入=1 持股=2(区间） 卖出=3 减仓=4 空仓=5(区间） 无信号=0
    85.     df['signal'] = 0
    86.     df['signal_reason'] = ''
    87.     # A. 股价向上突破定性线，定性线上行，买入。
    88.     # 翻译：收盘价涨跌幅为正 & 前一日收盘价小于定性价格 & 收盘价大于定性价格 & 定性线连续三日涨跌幅为正
    89.     df.loc[(df['close_rate']>0) & (df['closePrice'].shift(1)'ma20'].shift(1)) & (df['closePrice']>df['ma20']) & (df['ma20_direction']==1),'signal'] = 1
    90.     df.loc[(df['close_rate']>0) & (df['closePrice'].shift(1)'ma20'].shift(1)) & (df['closePrice']>df['ma20']) & (df['ma20_direction']==1),'signal_reason'] = 'A'
    91.     # B.定量线上穿定性线形成黄金交叉，买入。
    92.     # 翻译：前一日定量价格小于前一日定性价格 & 当日定量价格大于定性价格
    93.     df.loc[(df['ma5'].shift(1)'ma20'].shift(1)) & (df['ma5']>df['ma20']),'signal'] = 1
    94.     df.loc[(df['ma5'].shift(1)'ma20'].shift(1)) & (df['ma5']>df['ma20']),'signal_reason'] = 'B'
    95.     # C.股价下跌，遇定性线上行支撑止跌回升，买入。
    96.     # 翻译：连续三日收盘价涨跌幅为负 & 定性线连续三日涨跌幅为正 & 股价被定性线支撑 & 下一日收盘价正跌幅为正
    97.     df.loc[(df['closePrice_direction']==-1) & (df['ma20_direction']==1) & (df['close_ma20_distance']<=2) & (df['close_rate'].shift(-1)>0),'signal'] = 1
    98.     df.loc[(df['closePrice_direction']==-1) & (df['ma20_direction']==1) & (df['close_ma20_distance']<=2) & (df['close_rate'].shift(-1)>0),'signal_reason'] = 'C'
    99.     # D.定性线上行，股价在定性线上方向上突破定量线，买入。
    100.     # 翻译：定性线连续三日涨跌幅为正 & 收盘价大于定性线价格 & （前一日收盘价小于前一日定量价格 & 当日收盘价大于定量价格）
    101.     df.loc[(df['ma20_direction']==1) & (df['closePrice']>df['ma20']) & (df['closePrice'].shift(1)'ma5'].shift(1)) & (df['closePrice']>df['ma5']),'signal'] = 1
    102.     df.loc[(df['ma20_direction']==1) & (df['closePrice']>df['ma20']) & (df['closePrice'].shift(1)'ma5'].shift(1)) & (df['closePrice']>df['ma5']),'signal_reason'] = 'D'
    103.     # E.定量线下行，遇定性线上行支撑止跌后再度上升，买入。
    104.     # 翻译：连续三日定量线涨跌幅为负 & 连续三日定性线涨跌幅为正 & 定量线被定性线支撑 & 下一日定量线涨跌幅为正
    105.     df.loc[(df['ma5_direction']==-1) & (df['ma20_direction']==1) & (df['ma5_ma20_distance']<=2) & (df['ma5_rate'].shift(-1)>0),'signal'] = 1
    106.     df.loc[(df['ma5_direction']==-1) & (df['ma20_direction']==1) & (df['ma5_ma20_distance']<=2) & (df['ma5_rate'].shift(-1)>0),'signal_reason'] = 'E'
    107.     # F.股价、定量线、定性线多头排列，持股
    108.     # 翻译：收盘价大于定量价格大于定性价格
    109.     df.loc[(df['closePrice']>df['ma5']) & (df['ma5']>df['ma20']),'signal'] = 2
    110.     df.loc[(df['closePrice']>df['ma5']) & (df['ma5']>df['ma20']),'signal_reason'] = 'F'
    111.  
    112.     # 2） 卖出和空仓原则
    113.     # A.股价跌破定性线，定性线走平或已拐头下行，卖出
    114.     # 翻译：前一日收盘价大于定性价格 & 当日收盘价小于定性价格 & 定性价格涨跌幅小于等于0
    115.     df.loc[(df['closePrice'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma20']) & (df['ma20_rate']<=0),'signal'] = 3
    116.     df.loc[(df['closePrice'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma20']) & (df['ma20_rate']<=0),'signal_reason'] = 'A'
    117.     # B.多头排列期间，股价跌破定量线，减仓
    118.     # 翻译：前一日收盘价大于定量价格大于定性价格 & 当日收盘价小于定量价格
    119.     df.loc[(df['closePrice'].shift(1)>df['ma5'].shift(1)) & (df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma5']),'signal'] = 4
    120.     df.loc[(df['closePrice'].shift(1)>df['ma5'].shift(1)) & (df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma5']),'signal_reason'] = 'B'
    121.     # C.股价急速飙升，远离定性线，减仓
    122.     # 翻译：股价飙升 & 收盘价远离定性价格
    123.     df.loc[(df['fast_up']==1) & (df['close_ma5_distance']>=50),'signal'] = 4
    124.     df.loc[(df['fast_up']==1) & (df['close_ma5_distance']>=50),'signal_reason'] = 'C'
    125.     # D.定性线下行，空仓
    126.     # 翻译：连续三日定性价格涨跌幅为负
    127.     df.loc[df['ma20_direction']==-1,'signal'] = 5
    128.     df.loc[df['ma20_direction']==-1,'signal_reason'] = 'D'
    129.  
    130.     title_str = '双均线组合信号'
    131.     whole_header = ['日期','收盘价','开盘价','最高价','最低价','ma5','ma20']
    132.     whole_df = df
    133.     whole_pd_header = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice','ma5','ma20']
    134.     line_pd_header = ['ma5','ma20']
    135.     line_color_list = [0,2]
    136.  
    137.     line_data = {
    138.         'title_str':title_str,
    139.         'whole_header':whole_header,
    140.         'whole_df':whole_df,
    141.         'whole_pd_header':whole_pd_header,
    142.         'line_pd_header':line_pd_header,
    143.         'line_color_list':line_color_list
    144.     }
    145.     return line_data

    运行工具代码

     导入需要的包、K线控件、日期横坐标控件、分页表格控件

    1. import sys,json,os,math,time
    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')
    14.  
    15. class RotateAxisItem(pg.AxisItem):
    16.     def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):
    17.         p.setRenderHint(p.Antialiasing,False)
    18.         p.setRenderHint(p.TextAntialiasing,True)
    19.  
    20.         ## draw long line along axis
    21.         pen,p1,p2 = axisSpec
    22.         p.setPen(pen)
    23.         p.drawLine(p1,p2)
    24.         p.translate(0.5,0)  ## resolves some damn pixel ambiguity
    25.  
    26.         ## draw ticks
    27.         for pen,p1,p2 in tickSpecs:
    28.             p.setPen(pen)
    29.             p.drawLine(p1,p2)
    30.  
    31.         ## draw all text
    32.         # if self.tickFont is not None:
    33.         #     p.setFont(self.tickFont)
    34.         p.setPen(self.pen())
    35.         for rect,flags,text in textSpecs:
    36.             # this is the important part
    37.             p.save()
    38.             p.translate(rect.x(),rect.y())
    39.             p.rotate(-30)
    40.             p.drawText(-rect.width(),rect.height(),rect.width(),rect.height(),flags,text)
    41.             # restoring the painter is *required*!!!
    42.             p.restore()
    43.  
    44. class CandlestickItem(pg.GraphicsObject):
    45.     def __init__(self, data):
    46.         pg.GraphicsObject.__init__(self)
    47.         self.data = data  ## data must have fields: time, open, close, min, max
    48.         self.generatePicture()
    49.  
    50.     def generatePicture(self):
    51.         ## pre-computing a QPicture object allows paint() to run much more quickly,
    52.         ## rather than re-drawing the shapes every time.
    53.         self.picture = QtGui.QPicture()
    54.         p = QtGui.QPainter(self.picture)
    55.         p.setPen(pg.mkPen('d'))
    56.         w = (self.data[1][0] - self.data[0][0]) / 3.
    57.         for (t, open, close, min, max) in self.data:
    58.             p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))
    59.             if open < close:
    60.                 p.setBrush(pg.mkBrush('r'))
    61.             else:
    62.                 p.setBrush(pg.mkBrush('g'))
    63.             p.drawRect(QtCore.QRectF(t-w, open, w * 2, close - open))
    64.         p.end()
    65.  
    66.     def paint(self, p, *args):
    67.         p.drawPicture(0, 0, self.picture)
    68.  
    69.     def boundingRect(self):
    70.         ## boundingRect _must_ indicate the entire area that will be drawn on
    71.         ## or else we will get artifacts and possibly crashing.
    72.         ## (in this case, QPicture does all the work of computing the bouning rect for us)
    73.         return QtCore.QRectF(self.picture.boundingRect())
    74.  
    75. # 分页表格控件，单选
    76. class PageTableWidget(QtWidgets.QWidget):
    77.     def __init__(self):
    78.         super().__init__()
    79.         self.init_data()
    80.         self.init_ui()
    81.         pass
    82.  
    83.     def init_data(self):
    84.         # 表格全数据 二维数组
    85.         self.table_full_data: List[Any] = []
    86.         # 表格右键菜单 {菜单名：索引指向}
    87.         self.table_right_menus: Dict[str, str] = {}
    88.  
    89.         self.total_page_count: int = 0
    90.         self.total_rows_count: int = 0
    91.         self.current_page: int = 1
    92.         self.single_page_rows: int = 50
    93.         pass
    94.  
    95.     def init_ui(self):
    96.         pre_page_btn = QtWidgets.QPushButton('上一页')
    97.         pre_page_btn.clicked.connect(self.pre_page_btn_clicked)
    98.         next_page_btn = QtWidgets.QPushButton('下一页')
    99.         next_page_btn.clicked.connect(self.next_page_btn_clicked)
    100.  
    101.         tip_label_0 = QtWidgets.QLabel('第')
    102.         self.witch_page_lineedit = QtWidgets.QLineEdit()
    103.         self.int_validator = QtGui.QIntValidator()
    104.         self.witch_page_lineedit.setValidator(self.int_validator)
    105.         self.witch_page_lineedit.setMaximumWidth(20)
    106.         tip_label_1 = QtWidgets.QLabel('页')
    107.         go_page_btn = QtWidgets.QPushButton('前往')
    108.         go_page_btn.clicked.connect(self.go_page_btn_clicked)
    109.         layout_witch_page = QtWidgets.QHBoxLayout()
    110.         layout_witch_page.addWidget(tip_label_0)
    111.         layout_witch_page.addWidget(self.witch_page_lineedit)
    112.         layout_witch_page.addWidget(tip_label_1)
    113.         layout_witch_page.addWidget(go_page_btn)
    114.         layout_witch_page.addStretch(1)
    115.  
    116.         layout_pagechange = QtWidgets.QHBoxLayout()
    117.         layout_pagechange.addWidget(pre_page_btn)
    118.         layout_pagechange.addWidget(next_page_btn)
    119.         layout_pagechange.addLayout(layout_witch_page)
    120.  
    121.         self.total_page_count_label = QtWidgets.QLabel(f"共0页")
    122.         self.total_rows_count_label = QtWidgets.QLabel(f"共0行")
    123.         self.current_page_label = QtWidgets.QLabel(f"当前第0页")
    124.         layout_pagestatus = QtWidgets.QHBoxLayout()
    125.         layout_pagestatus.addWidget(self.total_page_count_label)
    126.         layout_pagestatus.addWidget(self.total_rows_count_label)
    127.         layout_pagestatus.addWidget(self.current_page_label)
    128.         layout_pagestatus.addStretch(1)
    129.  
    130.         layout_top = QtWidgets.QVBoxLayout()
    131.         layout_top.addLayout(layout_pagechange)
    132.         layout_top.addLayout(layout_pagestatus)
    133.  
    134.         self.target_table = QtWidgets.QTableWidget()
    135.         self.target_table.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
    136.         self.target_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
    137.  
    138.         layout = QtWidgets.QVBoxLayout()
    139.         layout.addLayout(layout_top)
    140.         layout.addWidget(self.target_table)
    141.         self.setLayout(layout)
    142.         pass
    143.  
    144.     def set_table_init_data(self, data: Dict[str, Any]):
    145.         '''设置表头、右键菜单等初始化内容'''
    146.         headers: List[str] = data['headers']
    147.  
    148.         self.target_table.setColumnCount(len(headers))
    149.         self.target_table.setHorizontalHeaderLabels(headers)
    150.         pass
    151.  
    152.     def set_table_full_data(self, data: List[Any]):
    153.         '''表格全数据'''
    154.         self.table_full_data = data
    155.         self.total_rows_count = len(data)
    156.         self.total_page_count = math.ceil(self.total_rows_count / self.single_page_rows)
    157.         self.current_page = 1
    158.  
    159.         self.int_validator.setRange(1, self.total_page_count)
    160.  
    161.         self.total_page_count_label.setText(f"共{self.total_page_count}页")
    162.         self.total_rows_count_label.setText(f"共{self.total_rows_count}行")
    163.         self.caculate_current_show_data()
    164.         pass
    165.  
    166.     def setting_current_pagestatus_label(self):
    167.         self.current_page_label.setText(f"当前第{self.current_page}页")
    168.         pass
    169.  
    170.     def caculate_current_show_data(self):
    171.         '''计算当前要显示的数据'''
    172.         start_dot = (self.current_page - 1) * self.single_page_rows
    173.         end_dot = start_dot + self.single_page_rows
    174.         current_data = self.table_full_data[start_dot:end_dot]
    175.         self.fill_table_content(current_data)
    176.         self.setting_current_pagestatus_label()
    177.         pass
    178.  
    179.     def fill_table_content(self, data: List[Any]):
    180.         self.target_table.clearContents()
    181.         self.target_table.setRowCount(len(data))
    182.         for r_i, r_v in enumerate(data):
    183.             for c_i, c_v in enumerate(r_v):
    184.                 cell = QtWidgets.QTableWidgetItem(str(c_v))
    185.                 self.target_table.setItem(r_i, c_i, cell)
    186.                 pass
    187.             pass
    188.         self.target_table.resizeColumnsToContents()
    189.         pass
    190.  
    191.     def go_page_btn_clicked(self):
    192.         '''前往按钮点击'''
    193.         input_page = self.witch_page_lineedit.text()
    194.         input_page = input_page.strip()
    195.         if not input_page:
    196.             QtWidgets.QMessageBox.information(
    197.                 self,
    198.                 '提示',
    199.                 '请输入要跳转的页码',
    200.                 QtWidgets.QMessageBox.Yes
    201.             )
    202.             return
    203.         input_page = int(input_page)
    204.         if input_page < 0 or input_page > self.total_page_count:
    205.             QtWidgets.QMessageBox.information(
    206.                 self,
    207.                 '提示',
    208.                 '输入的页码超出范围',
    209.                 QtWidgets.QMessageBox.Yes
    210.             )
    211.             return
    212.         self.current_page = input_page
    213.         self.caculate_current_show_data()
    214.         pass
    215.  
    216.     def pre_page_btn_clicked(self):
    217.         '''上一页按钮点击'''
    218.         if self.current_page <= 1:
    219.             QtWidgets.QMessageBox.information(
    220.                 self,
    221.                 '提示',
    222.                 '已经是首页',
    223.                 QtWidgets.QMessageBox.Yes
    224.             )
    225.             return
    226.         self.current_page -= 1
    227.         self.caculate_current_show_data()
    228.         pass
    229.  
    230.     def next_page_btn_clicked(self):
    231.         '''下一页按钮点击'''
    232.         if self.current_page >= self.total_page_count:
    233.             QtWidgets.QMessageBox.information(
    234.                 self,
    235.                 '提示',
    236.                 '已经是最后一页',
    237.                 QtWidgets.QMessageBox.Yes
    238.             )
    239.             return
    240.         self.current_page += 1
    241.         self.caculate_current_show_data()
    242.         pass
    243.  
    244.     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.table_header = ['日期','信号','未复权收盘价','解说']
    9.         self.please_select_str = '---请选择---'
    10.         self.func_map = {
    11.             '标记多头排列区间':'a',
    12.             '标记空头排列区间':'b',
    13.             '20日线斜率为正':'c'
    14.         }
    15.         self.func_item_list = []
    16.         self.duration_map = {
    17.             '今年':'a',
    18.             '最近一年':'b',
    19.             '最近两年':'c'
    20.         }
    21.         # https://www.sioe.cn/yingyong/yanse-rgb-16/
    22.         self.color_line = (30, 144, 255)
    23.         # 0 幽灵的白色; 1 纯黄; 2 紫红色; 3 纯绿; 4 道奇蓝
    24.         self.color_list = [(248,248,255),(255,255,0),(255,0,255),(0,128,0),(30,144,255)]
    25.         self.buy_color = (220, 20, 60)  # 猩红
    26.         self.hold_color = (255, 140, 0)  # 深橙色
    27.         self.sell_color = (50, 205, 50)  # 酸橙绿
    28.         self.subpos_color = (24, 252, 0)  # 草坪绿
    29.         self.emptypos_color = (0, 128, 0)  # 纯绿
    30.         self.main_fixed_target_list = []  # 主体固定曲线，不能被删除
    31.         self.whole_df = None
    32.         self.whole_header = None
    33.         self.whole_pd_header = None
    34.         self.current_whole_data = None
    35.         self.current_whole_df = None
    36.         self.line_pd_header = []
    37.         self.line_color_list = []
    38.         self.signal_show_list = []
    39.         pass
    40.     def init_ui(self):
    41.         # 控制面板 start
    42.         left_tip = QtWidgets.QLabel('左边界')
    43.         self.left_point = QtWidgets.QDateEdit()
    44.         self.left_point.setDisplayFormat('yyyy-MM-dd')
    45.         self.left_point.setCalendarPopup(True)
    46.         right_tip = QtWidgets.QLabel('右边界')
    47.         self.right_point = QtWidgets.QDateEdit()
    48.         self.right_point.setDisplayFormat('yyyy-MM-dd')
    49.         self.right_point.setCalendarPopup(True)
    50.         duration_sel_btn = QtWidgets.QPushButton('确定')
    51.         duration_sel_btn.clicked.connect(self.duration_sel_btn_clicked)
    52.         duration_reset_btn = QtWidgets.QPushButton('重置')
    53.         duration_reset_btn.clicked.connect(self.duration_reset_btn_clicked)
    54.  
    55.         self.whole_duration_label = QtWidgets.QLabel('原始最宽边界：左边界~右边界')
    56.         self.now_duration_label = QtWidgets.QLabel('当前显示最宽边界：左边界~右边界')
    57.  
    58.         self.buy_checkbox = QtWidgets.QCheckBox('买入')
    59.         self.buy_checkbox.stateChanged.connect(lambda: self.signal_checkbox_state(self.buy_checkbox))
    60.         self.hold_checkbox = QtWidgets.QCheckBox('持股')
    61.         self.hold_checkbox.stateChanged.connect(lambda: self.signal_checkbox_state(self.hold_checkbox))
    62.         self.sell_checkbox = QtWidgets.QCheckBox('卖出')
    63.         self.sell_checkbox.stateChanged.connect(lambda: self.signal_checkbox_state(self.sell_checkbox))
    64.         self.subpos_checkbox = QtWidgets.QCheckBox('减仓')
    65.         self.subpos_checkbox.stateChanged.connect(lambda: self.signal_checkbox_state(self.subpos_checkbox))
    66.         self.emptypos_checkbox = QtWidgets.QCheckBox('空仓')
    67.         self.emptypos_checkbox.stateChanged.connect(lambda: self.signal_checkbox_state(self.emptypos_checkbox))
    68.  
    69.         layout_date = QtWidgets.QHBoxLayout()
    70.         layout_date.addWidget(left_tip)
    71.         layout_date.addWidget(self.left_point)
    72.         layout_date.addWidget(right_tip)
    73.         layout_date.addWidget(self.right_point)
    74.         layout_date.addWidget(duration_sel_btn)
    75.         layout_date.addWidget(duration_reset_btn)
    76.         layout_date.addWidget(self.buy_checkbox)
    77.         layout_date.addWidget(self.hold_checkbox)
    78.         layout_date.addWidget(self.sell_checkbox)
    79.         layout_date.addWidget(self.subpos_checkbox)
    80.         layout_date.addWidget(self.emptypos_checkbox)
    81.         layout_date.addStretch(1)
    82.         layout_duration = QtWidgets.QHBoxLayout()
    83.         layout_duration.addWidget(self.whole_duration_label)
    84.         layout_duration.addSpacing(30)
    85.         layout_duration.addWidget(self.now_duration_label)
    86.         layout_duration.addStretch(1)
    87.         # 控制面板 end
    88.  
    89.         self.title_label = QtWidgets.QLabel('执行过程查看')
    90.         self.title_label.setAlignment(Qt.AlignCenter)
    91.         self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')
    92.  
    93.         xax = RotateAxisItem(orientation='bottom')
    94.         xax.setHeight(h=80)
    95.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    96.         self.pw.setMouseEnabled(x=True, y=True)
    97.         # self.pw.enableAutoRange(x=False,y=True)
    98.         self.pw.setAutoVisible(x=False, y=True)
    99.  
    100.         self.table = PageTableWidget()
    101.         self.table.set_table_init_data({'headers':self.table_header})
    102.  
    103.         layout_down = QtWidgets.QHBoxLayout()
    104.         layout_down.addWidget(self.pw,6)
    105.         layout_down.addWidget(self.table,1)
    106.  
    107.         layout = QtWidgets.QVBoxLayout()
    108.         layout.addWidget(self.title_label)
    109.         layout.addLayout(layout_date)
    110.         layout.addLayout(layout_duration)
    111.         layout.addLayout(layout_down)
    112.         self.setLayout(layout)
    113.         pass
    114.     def set_data(self,data:Dict[str,Any]):
    115.         title_str = data['title_str']
    116.         whole_header = data['whole_header']
    117.         whole_df = data['whole_df']
    118.         whole_pd_header = data['whole_pd_header']
    119.         line_pd_header = data['line_pd_header']
    120.         line_color_list = data['line_color_list']
    121.  
    122.         self.whole_header = whole_header
    123.         self.whole_df = whole_df
    124.         self.whole_pd_header = whole_pd_header
    125.         self.line_pd_header = line_pd_header
    126.         self.line_color_list = line_color_list
    127.  
    128.         self.title_label.setText(title_str)
    129.         self.whole_duration_label.setText(f"原始最宽边界：{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")
    130.  
    131.         self.current_whole_df = self.whole_df.copy()
    132.         self.caculate_and_show_data()
    133.         pass
    134.     def caculate_and_show_data(self):
    135.         df = self.current_whole_df.copy()
    136.         df.reset_index(inplace=True)
    137.         df['i_count'] = [i for i in range(len(df))]
    138.         tradeDate_list = df['tradeDate'].values.tolist()
    139.         x = range(len(df))
    140.         xTick_show = []
    141.         x_dur = math.ceil(len(df)/20)
    142.         for i in range(0,len(df),x_dur):
    143.             xTick_show.append((i,tradeDate_list[i]))
    144.         if len(df)%20 != 0:
    145.             xTick_show.append((len(df)-1,tradeDate_list[-1]))
    146.         candle_data = []
    147.         for i,row in df.iterrows():
    148.             candle_data.append((row['i_count'],row['openPrice'],row['closePrice'],row['lowestPrice'],row['highestPrice']))
    149.  
    150.         self.current_whole_data = df.loc[:,self.whole_pd_header].values.tolist()
    151.         # 开始配置显示的内容
    152.         self.pw.clear()
    153.         self.func_item_list.clear()
    154.         self.nocheck_checkbox()
    155.         self.remove_signal_show()
    156.  
    157.         self.now_duration_label.setText(f"当前显示最宽边界：{df.iloc[0]['tradeDate']}~{df.iloc[-1]['tradeDate']}")
    158.  
    159.         xax = self.pw.getAxis('bottom')
    160.         xax.setTicks([xTick_show])
    161.  
    162.         candle_fixed_target = CandlestickItem(candle_data)
    163.         self.main_fixed_target_list.append(candle_fixed_target)
    164.         self.pw.addItem(candle_fixed_target)
    165.  
    166.         # 曲线
    167.         if len(self.line_pd_header)>0:
    168.             for i,item in enumerate(self.line_pd_header):
    169.                 line_fixed_target = pg.PlotCurveItem(x=np.array(x), y=np.array(df[item].values.tolist()),
    170.                                             pen=pg.mkPen({'color': self.color_list[self.line_color_list[i]], 'width': 2}),
    171.                                             connect='finite')
    172.                 self.main_fixed_target_list.append(line_fixed_target)
    173.                 self.pw.addItem(line_fixed_target)
    174.             pass
    175.  
    176.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    177.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    178.         self.label = pg.TextItem()
    179.  
    180.         self.pw.addItem(self.vLine, ignoreBounds=True)
    181.         self.pw.addItem(self.hLine, ignoreBounds=True)
    182.         self.pw.addItem(self.label, ignoreBounds=True)
    183.  
    184.         self.vb = self.pw.getViewBox()
    185.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    186.         self.pw.enableAutoRange()
    187.         pass
    188.  
    189.     def mouseMoved(self,evt):
    190.         pos = evt[0]
    191.         if self.pw.sceneBoundingRect().contains(pos):
    192.             mousePoint = self.vb.mapSceneToView(pos)
    193.             index = int(mousePoint.x())
    194.             if index>=0 and index<len(self.current_whole_data):
    195.                 target_data = self.current_whole_data[index]
    196.                 html_str = ''
    197.                 for i,item in enumerate(self.whole_header):
    198.                     html_str += f"
      {item}:{target_data[i]}"
    199.                 self.label.setHtml(html_str)
    200.                 self.label.setPos(mousePoint.x(),mousePoint.y())
    201.             self.vLine.setPos(mousePoint.x())
    202.             self.hLine.setPos(mousePoint.y())
    203.         pass
    204.     def mouseClicked(self,evt):
    205.         pass
    206.     def updateViews(self):
    207.         pass
    208.  
    209.     def duration_sel_btn_clicked(self):
    210.         '''边界选择'''
    211.         left_point = self.left_point.date().toString('yyyy-MM-dd')
    212.         right_point = self.right_point.date().toString('yyyy-MM-dd')
    213.         df = self.whole_df.copy()
    214.         df['o_date'] = pd.to_datetime(df['tradeDate'])
    215.         self.current_whole_df = df.loc[(df['o_date']>=left_point) & (df['o_date']<=right_point)].copy()
    216.         self.caculate_and_show_data()
    217.         pass
    218.     def duration_reset_btn_clicked(self):
    219.         '''边界重置'''
    220.         self.current_whole_df = self.whole_df.copy()
    221.         self.caculate_and_show_data()
    222.         pass
    223.     def nocheck_checkbox(self):
    224.         self.buy_checkbox.setChecked(False)
    225.         self.hold_checkbox.setChecked(False)
    226.         self.sell_checkbox.setChecked(False)
    227.         self.subpos_checkbox.setChecked(False)
    228.         self.emptypos_checkbox.setChecked(False)
    229.         pass
    230.     def remove_signal_show(self):
    231.         for item in self.signal_show_list:
    232.             self.pw.removeItem(item)
    233.         self.signal_show_list.clear()
    234.         pass
    235.     def signal_checkbox_state(self,cb):
    236.         self.remove_signal_show()
    237.         signal_list = []
    238.         if self.buy_checkbox.isChecked():
    239.             signal_list.append(1)
    240.         if self.hold_checkbox.isChecked():
    241.             signal_list.append(2)
    242.         if self.sell_checkbox.isChecked():
    243.             signal_list.append(3)
    244.         if self.subpos_checkbox.isChecked():
    245.             signal_list.append(4)
    246.         if self.emptypos_checkbox.isChecked():
    247.             signal_list.append(5)
    248.  
    249.         if len(signal_list)<=0:
    250.             # 删除所有的信号
    251.             self.remove_signal_show()
    252.             pass
    253.         else:
    254.             df = self.current_whole_df.copy()
    255.             df.reset_index(inplace=True)
    256.             if 1 in signal_list:
    257.                 # 买 220,20,60  猩红
    258.                 df_buy = df.loc[df['signal']==1].copy()
    259.                 for i, row in df_buy.iterrows():
    260.                     buy_fixed_target = pg.InfiniteLine(pos=(i, 0), movable=False, angle=90,
    261.                                                        pen=pg.mkPen({'color': self.buy_color, 'width': 2}),
    262.                                                        label=str(row['close']),
    263.                                                        labelOpts={'position': 0.05, 'color': self.buy_color,
    264.                                                                   'movable': True, 'fill': (
    265.                                                            self.buy_color[0], self.buy_color[1], self.buy_color[2], 30)})
    266.                     self.signal_show_list.append(buy_fixed_target)
    267.                     self.pw.addItem(buy_fixed_target)
    268.                     pass
    269.                 pass
    270.             if 2 in signal_list:
    271.                 # 持股 255,140,0 深橙色
    272.                 df['ext_hold_count'] = [i for i in range(len(df))]
    273.                 df['ext_hold_0'] = 0
    274.                 df.loc[df['signal'] == 2, 'ext_hold_0'] = 1
    275.                 # start 值
    276.                 df['ext_hold_1'] = df['ext_hold_0'] - df['ext_hold_0'].shift(1)
    277.                 # end 值
    278.                 df['ext_hold_2'] = df['ext_hold_0'] - df['ext_hold_0'].shift(-1)
    279.                 df_hold_start = df.loc[df['ext_hold_1'] == 1].copy()
    280.                 df_hold_end = df.loc[df['ext_hold_2'] == 1].copy()
    281.                 hold_start_count_list = df_hold_start['ext_hold_count'].values.tolist()
    282.                 hold_end_count_list = df_hold_end['ext_hold_count'].values.tolist()
    283.                 if hold_end_count_list[0] < hold_start_count_list[0]:
    284.                     hold_end_count_list = hold_end_count_list[1:]
    285.                 if hold_end_count_list[-1] < hold_start_count_list[-1]:
    286.                     hold_start_count_list = hold_start_count_list[:-1]
    287.                 for i, item in enumerate(hold_end_count_list):
    288.                     lr = pg.LinearRegionItem([hold_start_count_list[i], item], movable=False, brush=(
    289.                     self.hold_color[0], self.hold_color[1], self.hold_color[2], 100))
    290.                     lr.setZValue(-100)
    291.                     self.signal_show_list.append(lr)
    292.                     self.pw.addItem(lr)
    293.                     pass
    294.                 pass
    295.             if 3 in signal_list:
    296.                 # 卖出 50,205,50 酸橙绿
    297.                 df_sell = df.loc[df['signal'] == 3].copy()
    298.                 for i, row in df_sell.iterrows():
    299.                     sell_fixed_target = pg.InfiniteLine(pos=(i, 0), movable=False, angle=90,
    300.                                                        pen=pg.mkPen({'color': self.sell_color, 'width': 2}),
    301.                                                        label=str(row['close']),
    302.                                                        labelOpts={'position': 0.05, 'color': self.sell_color,
    303.                                                                   'movable': True, 'fill': (
    304.                                                                self.sell_color[0], self.sell_color[1], self.sell_color[2],
    305.                                                                30)})
    306.                     self.signal_show_list.append(sell_fixed_target)
    307.                     self.pw.addItem(sell_fixed_target)
    308.                     pass
    309.                 pass
    310.             if 4 in signal_list:
    311.                 # 减仓 24,252,0 草坪绿
    312.                 df_subpos = df.loc[df['signal'] == 4].copy()
    313.                 for i, row in df_subpos.iterrows():
    314.                     subpos_fixed_target = pg.InfiniteLine(pos=(i, 0), movable=False, angle=90,
    315.                                                         pen=pg.mkPen({'color': self.subpos_color, 'width': 2}),
    316.                                                         label=str(row['close']),
    317.                                                         labelOpts={'position': 0.05, 'color': self.subpos_color,
    318.                                                                    'movable': True, 'fill': (
    319.                                                                 self.subpos_color[0], self.subpos_color[1],
    320.                                                                 self.subpos_color[2],
    321.                                                                 30)})
    322.                     self.signal_show_list.append(subpos_fixed_target)
    323.                     self.pw.addItem(subpos_fixed_target)
    324.                     pass
    325.                 pass
    326.             if 5 in signal_list:
    327.                 # 空仓 0,128,0 纯绿
    328.                 df['ext_empty_count'] = [i for i in range(len(df))]
    329.                 df['ext_empty_0'] = 0
    330.                 df.loc[df['signal']==5,'ext_empty_0'] = 1
    331.                 # start 值
    332.                 df['ext_empty_1'] = df['ext_empty_0']-df['ext_empty_0'].shift(1)
    333.                 # end 值
    334.                 df['ext_empty_2'] = df['ext_empty_0']-df['ext_empty_0'].shift(-1)
    335.                 df_empty_start = df.loc[df['ext_empty_1']==1].copy()
    336.                 df_empty_end = df.loc[df['ext_empty_2']==1].copy()
    337.                 empty_start_count_list = df_empty_start['ext_empty_count'].values.tolist()
    338.                 empty_end_count_list = df_empty_end['ext_empty_count'].values.tolist()
    339.                 if empty_end_count_list[0]0]:
    340.                     empty_end_count_list = empty_end_count_list[1:]
    341.                 if empty_end_count_list[-1]1]:
    342.                     empty_start_count_list = empty_start_count_list[:-1]
    343.                 for i,item in enumerate(empty_end_count_list):
    344.                     lr = pg.LinearRegionItem([empty_start_count_list[i],item],movable=False,brush=(self.emptypos_color[0],self.emptypos_color[1],self.emptypos_color[2],100))
    345.                     lr.setZValue(-100)
    346.                     self.signal_show_list.append(lr)
    347.                     self.pw.addItem(lr)
    348.                     pass
    349.                 pass
    350.             df['signal_name'] = ''
    351.             for item in signal_list:
    352.                 signal_name = ''
    353.                 if item == 1:
    354.                     signal_name = '买入'
    355.                 if item == 2:
    356.                     signal_name = '持股'
    357.                 if item == 3:
    358.                     signal_name = '卖出'
    359.                 if item == 4:
    360.                     signal_name = '减仓'
    361.                 if item == 5:
    362.                     signal_name = '空仓'
    363.                 df.loc[df['signal'] == item,'signal_name'] = signal_name
    364.             df_table = df.loc[df['signal'].isin(signal_list)].copy()
    365.             table_data = df_table.loc[:, ['tradeDate', 'signal_name', 'close','signal_reason']].values.tolist()
    366.             self.table.set_table_full_data(table_data)
    367.         pass
    368.     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.please_select_str: str = '--请选择--'
    18.         self.stratege_name_list: List = []
    19.         self.tip_msg_0: str = '1.选择策略所在文件夹；2.选择策略；3.点击运行。'
    20.         self.stratege_path: str = ''
    21.         self.stratege_run_start_time = None
    22.         self.stratege_start = False
    23.         self.current_stratege_py_str: str = ''
    24.         pass
    25.     def init_ui(self):
    26.         self.setWindowTitle('策略运行小工具')
    27.         tip_0 = QtWidgets.QLabel('选择策略所在文件夹:')
    28.         self.stratege_dir_lineedit = QtWidgets.QLineEdit()
    29.         self.stratege_dir_lineedit.setReadOnly(True)
    30.         stratege_choice_btn = QtWidgets.QPushButton('选择文件夹')
    31.         stratege_choice_btn.clicked.connect(self.stratege_choice_btn_clicked)
    32.  
    33.         tip_1 = QtWidgets.QLabel('策略:')
    34.         self.stratege_combox = QtWidgets.QComboBox()
    35.         self.stratege_combox.addItem(self.please_select_str)
    36.         self.stratege_combox.currentIndexChanged.connect(self.stratege_combox_currentIndexChanged)
    37.  
    38.         self.run_btn = QtWidgets.QPushButton('运行')
    39.         self.run_btn.clicked.connect(self.run_btn_clicked)
    40.         self.force_stop_btn = QtWidgets.QPushButton('强制停止')
    41.         self.force_stop_btn.clicked.connect(self.force_stop_btn_clicked)
    42.  
    43.         layout_top_left = QtWidgets.QGridLayout()
    44.         layout_top_left.addWidget(tip_0,0,0,1,1)
    45.         layout_top_left.addWidget(self.stratege_dir_lineedit,0,1,1,3)
    46.         layout_top_left.addWidget(stratege_choice_btn,0,4,1,1)
    47.         layout_top_left.addWidget(tip_1,1,0,1,1)
    48.         layout_top_left.addWidget(self.stratege_combox,1,1,1,2)
    49.         layout_top_left.addWidget(self.run_btn,1,3,1,1)
    50.         layout_top_left.addWidget(self.force_stop_btn,1,4,1,1)
    51.  
    52.         self.tip_msg_label = QtWidgets.QLabel()
    53.         self.tip_msg_label.setWordWrap(True)
    54.         self.tip_msg_label.setText(self.tip_msg_0)
    55.  
    56.         layout_top = QtWidgets.QHBoxLayout()
    57.         layout_top.addLayout(layout_top_left,3)
    58.         layout_top.addWidget(self.tip_msg_label,1)
    59.  
    60.         self.code_textedit = QtWidgets.QTextEdit()
    61.         self.code_textedit.setReadOnly(True)
    62.  
    63.         layout = QtWidgets.QVBoxLayout()
    64.         layout.addLayout(layout_top)
    65.         layout.addWidget(self.code_textedit)
    66.         self.setLayout(layout)
    67.         pass
    68.     def register_event(self):
    69.         self.signal_runcode.connect(self.thread_run_excuted)
    70.         self.signal_time.connect(self.thread_time_excuted)
    71.         pass
    72.     def stratege_choice_btn_clicked(self):
    73.         '''选择策略所在文件夹'''
    74.         path = QtWidgets.QFileDialog.getExistingDirectory(
    75.             self,
    76.             '打开策略所在文件夹',
    77.             self.pre_output_dir
    78.         )
    79.         if not path:
    80.             return
    81.         self.stratege_path = path
    82.         self.stratege_dir_lineedit.setText(path)
    83.         file_list = os.listdir(path)
    84.         temp_file_list = set(self.stratege_name_list)
    85.         for item in file_list:
    86.             if item.endswith('.py'):
    87.                 temp_file_list.add(item)
    88.         self.stratege_name_list = list(temp_file_list)
    89.  
    90.         self.stratege_combox.clear()
    91.         self.stratege_combox.addItem(self.please_select_str)
    92.         self.stratege_combox.addItems(self.stratege_name_list)
    93.         pass
    94.     def stratege_combox_currentIndexChanged(self,cur_i:int):
    95.         cur_txt = self.stratege_combox.currentText()
    96.         if not cur_txt or cur_txt == self.please_select_str:
    97.             self.code_textedit.clear()
    98.             return
    99.         file_path = self.stratege_path + os.path.sep + cur_txt
    100.         with open(file_path,'r',encoding='utf-8') as fr:
    101.             code_txt = fr.read()
    102.         self.code_textedit.setPlainText(code_txt)
    103.         pass
    104.     def run_btn_clicked(self):
    105.         '''运行按钮'''
    106.         py_str = self.code_textedit.toPlainText()
    107.         if len(py_str)<10:
    108.             QtWidgets.QMessageBox.information(
    109.                 self,
    110.                 '提示',
    111.                 '请选择要执行的策略',
    112.                 QtWidgets.QMessageBox.Yes
    113.             )
    114.             return
    115.         self.current_stratege_py_str = py_str
    116.         self.run_btn.setDisabled(True)
    117.         self.stratege_combox.setDisabled(True)
    118.         self.stratege_run_start_time = datetime.now()
    119.         self.stratege_start = True
    120.  
    121.         if self.thread_run:
    122.             QtWidgets.QMessageBox.information(
    123.                 self,
    124.                 '提示',
    125.                 '有策略正在运行',
    126.                 QtWidgets.QMessageBox.Yes
    127.             )
    128.             return
    129.         self.thread_run = Thread(
    130.             target=self.running_run_thread,
    131.             args=(py_str,)
    132.         )
    133.         self.thread_run.start()
    134.         self.thread_time = Thread(
    135.             target=self.running_time_thread
    136.         )
    137.         self.thread_time.start()
    138.         pass
    139.     def force_stop_btn_clicked(self):
    140.         '''强制停止按钮'''
    141.         self.thread_run = None
    142.         self.thread_time = None
    143.  
    144.         self.run_btn.setDisabled(False)
    145.         self.stratege_combox.setDisabled(False)
    146.         self.stratege_start = False
    147.         pass
    148.     def running_run_thread(self,data:str):
    149.         '''执行代码线程'''
    150.         namespace = {}
    151.         fun_stragegy = compile(data,'','exec')
    152.         exec(fun_stragegy,namespace)
    153.         ret = namespace['excute_strategy']()
    154.         self.signal_runcode.emit(ret)
    155.         pass
    156.     def running_time_thread(self):
    157.         '''计时线程'''
    158.         while self.stratege_start:
    159.             now = datetime.now()
    160.             interval_time = (now-self.stratege_run_start_time).seconds
    161.             res_map = {'res':interval_time}
    162.             self.signal_time.emit(res_map)
    163.             time.sleep(1)
    164.         pass
    165.     def thread_run_excuted(self,data:Dict):
    166.         '''策略代码执行返回结果'''
    167.         self.run_btn.setDisabled(False)
    168.         self.stratege_combox.setDisabled(False)
    169.  
    170.         if not self.running_graph_widget:
    171.             self.running_graph_widget = PyQtGraphRunningWidget()
    172.         self.running_graph_widget.set_data(data)
    173.         self.running_graph_widget.showMaximized()
    174.         self.thread_run = None
    175.         self.thread_time = None
    176.         self.stratege_start = False
    177.  
    178.         QtWidgets.QMessageBox.information(
    179.             self,
    180.             '提示',
    181.             '当前策略运行完毕',
    182.             QtWidgets.QMessageBox.Yes
    183.         )
    184.         pass
    185.     def thread_time_excuted(self,data:Dict):
    186.         '''计时返回结果'''
    187.         res = data['res']
    188.         self.tip_msg_label.setText(f"{res}s")
    189.         pass
    190.     def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
    191.         if self.thread_time:
    192.             self.thread_time.join()
    193.         if self.thread_run:
    194.             self.thread_run.join()
    195.         if self.running_graph_widget:
    196.             self.running_graph_widget.close()
    197.         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

     运行，选择策略py文件所在目录，选择要执行的策略对应的py文件，点击运行

     策略执行完毕后，会弹出新的窗体显示策略执行结果图表，在该图表窗体中可以选择指定日期期间，分别查看各个信号的情况

    验证策略思想

    由于双均线涉及的交易原则太多，本文以验证“ 1）买入和持仓原则中的 A”为例，讲述工具验证策略思想的过程

    从右侧表格中可以看出 2020-09-10 的买入信号是通过买入交易原则 A 获得的，该交易原则的描述是“股价向上突破定性线，定性线上行，买入。”，查看左侧图，可以看出与描述相符

    优化策略

     优化策略是一个很细的活，这里以“1）买入和持仓原则中的 B”为例，讲述策略优化的过程。

    右侧表格中总共出现了两次B, 一次是2020-04-23， 另一次是2020-10-14；放大看左侧图

    2020-04-23这次的黄金交叉在次日就变为死亡交叉 ，这种黄金交叉有效时长太短暂，可以认为无效

    2020-10-14这次的黄金交叉也和上面的一样，太短暂，可以当做无效

    优化策略，增加自定义规则：

    描述代码逻辑（自定义）
    黄金交叉前一日定量价格小于定性价格，当日定量价格大于定性价格，后续两日定量价格都大于定性价格
    死亡交叉前一日定量价格大于定性价格，当日定量价格小于定性价格，后续两日定量价格都小于定性价格

     修改策略代码：

    在实现原则的代码前加入

    1.  # 黄金交叉
    2.     df['gold_x'] = 0
    3.     df.loc[(df['ma5'].shift(1)'ma20'].shift(1)) & (df['ma5']>df['ma20']) & (df['ma5'].shift(-1)>df['ma20'].shift(-1)) & (df['ma5'].shift(-2)>df['ma20'].shift(-2)),'gold_x'] = 1
    4.     # 死亡交叉
    5.     df['dead_x'] = 0
    6.     df.loc[(df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['ma5']'ma20']) & (df['ma5'].shift(-1)'ma20'].shift(-1)) & (df['ma5'].shift(-2)'ma20'].shift(-2)),'dead_x'] = 1

    将买入B原则的代码修改为

    1. # B.定量线上穿定性线形成黄金交叉，买入。
    2.     # 翻译：黄金交叉
    3.     df.loc[df['gold_x']==1,'signal'] = 1
    4.     df.loc[df['gold_x']==1,'signal_reason'] = 'B'

     修正后完整的策略代码为

    1. def excute_strategy():
    2.     '''
    3.     前置自定义规则：
    4.     A. 支撑和阻碍，定义为股价与均线价格之间的差值小于等于当前股价的2%
    5.     B. 股价下跌，连续三日收盘价涨跌幅为负
    6.     C. 均线下行，连续三日均线涨跌幅为负
    7.     D. 股价急速飙升，连续三日股价涨跌幅都大于等于5%
    8.     E. 远离，定义为股价与均线价格之间的差值大于等于当前股价的50%
    9.     1) 买入和持仓原则：
    10.     A. 股价向上突破定性线，定性线上行，买入。
    11.     翻译：收盘价涨跌幅为正 & 前一日收盘价小于定性价格 & 收盘价大于定性价格 & 定性线连续三日涨跌幅为正
    12.     B. 定量线上穿定性线形成黄金交叉，买入。
    13.     翻译：前一日定量价格小于前一日定性价格 & 当日定量价格大于定性价格
    14.     C. 股价下跌，遇定性线上行支撑止跌回升，买入。
    15.     翻译：连续三日收盘价涨跌幅为负 & 定性线连续三日涨跌幅为正 & 股价被定性线支撑 & 下一日收盘价正跌幅为正
    16.     D. 定性线上行，股价在定性线上方向上突破定量线，买入。
    17.     翻译：定性线连续三日涨跌幅为正 & 收盘价大于定性线价格 & （前一日收盘价小于前一日定量价格 & 当日收盘价大于定量价格）
    18.     E. 定量线下行，遇定性线上行支撑止跌后再度上升，买入。
    19.     翻译：连续三日定量线涨跌幅为负 & 连续三日定性线涨跌幅为正 & 定量线被定性线支撑 & 下一日定量线涨跌幅为正
    20.     F. 股价、定量线、定性线多头排列，持股
    21.     翻译：收盘价大于定量价格大于定性价格
    22.     2） 卖出和空仓原则
    23.     A. 股价跌破定性线，定性线走平或已拐头下行，卖出
    24.     翻译：前一日收盘价大于定性价格 & 当日收盘价小于定性价格 & 定性价格涨跌幅小于等于0
    25.     B. 多头排列期间，股价跌破定量线，减仓
    26.     翻译：前一日收盘价大于定量价格大于定性价格 & 当日收盘价小于定量价格
    27.     C. 股价急速飙升，远离定性线，减仓
    28.     翻译：股价飙升 & 收盘价远离定性价格
    29.     D. 定性线下行，空仓
    30.     翻译：连续三日定性价格涨跌幅为负
    31.     :return:
    32.     '''
    33.     import pandas as pd
    34.     import talib
    35.     df = pd.read_csv('E:/temp005/600660.csv',encoding='utf-8')
    36.     # 删除停牌的数据
    37.     df = df.loc[df['openPrice'] > 0].copy()
    38.     df['o_date'] = df['tradeDate']
    39.     df['o_date'] = pd.to_datetime(df['o_date'])
    40.     df = df.loc[df['o_date']>='2010-01-01'].copy()
    41.     # 保存未复权收盘价数据
    42.     df['close'] = df['closePrice']
    43.     # 计算前复权数据
    44.     df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    45.     df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    46.     df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    47.     df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    48.  
    49.     # 双均线组合
    50.     # 定性线：20日均线  定量线：5日均线
    51.     df['ma5'] = talib.MA(df['closePrice'],timeperiod=5)
    52.     df['ma20'] = talib.MA(df['closePrice'],timeperiod=20)
    53.     # 定性线涨跌幅
    54.     df['ma20_rate'] = df['ma20']-df['ma20'].shift(1)
    55.     # 定性线上行还是下行:连续三天涨跌幅为正，为上行，值为1；连续三天涨跌幅为负，为下行，值为-1
    56.     df['ma20_direction'] = 0
    57.     df.loc[(df['ma20_rate']>0) & (df['ma20_rate'].shift(1)>0) & (df['ma20_rate'].shift(2)>0),'ma20_direction'] = 1
    58.     df.loc[(df['ma20_rate']<0) & (df['ma20_rate'].shift(1)<0) & (df['ma20_rate'].shift(2)<0),'ma20_direction'] = -1
    59.     # 定量线涨跌幅
    60.     df['ma5_rate'] = df['ma5'] - df['ma5'].shift(1)
    61.     df['ma5_direction'] = 0
    62.     df.loc[(df['ma5_rate']>0) & (df['ma5_rate'].shift(1)>0) & (df['ma5_rate'].shift(2)>0),'ma5_direction'] = 1
    63.     df.loc[(df['ma5_rate']<0) & (df['ma5_rate'].shift(1)<0) & (df['ma5_rate'].shift(2)<0),'ma5_direction'] = -1
    64.     # 收盘价涨跌幅
    65.     df['close_rate'] = df['closePrice'] - df['closePrice'].shift(1)
    66.     df['closePrice_direction'] = 0
    67.     df.loc[(df['close_rate'] > 0) & (df['close_rate'].shift(1) > 0) & (df['close_rate'].shift(2) > 0), 'closePrice_direction'] = 1
    68.     df.loc[(df['close_rate'] < 0) & (df['close_rate'].shift(1) < 0) & (df['close_rate'].shift(2) < 0), 'closePrice_direction'] = -1
    69.     # 收盘价到定性线之间的距离
    70.     df['close_ma20_distance'] = (abs(df['closePrice']-df['ma20'])/df['closePrice'])*100
    71.     # 收盘价到定量线之间的距离
    72.     df['close_ma5_distance'] = (abs(df['closePrice']-df['ma5'])/df['closePrice'])*100
    73.     # 定量线到定性线之间的距离
    74.     df['ma5_ma20_distance'] = (abs(df['ma5']-df['ma20'])/df['ma5'])*100
    75.     # 定性线到定量线之间的距离
    76.     df['ma20_ma5_distance'] = (abs(df['ma5']-df['ma20'])/df['ma20'])*100
    77.     # 股价是否飙升
    78.     df['pre_fast_up'] = 0
    79.     df.loc[(df['close_rate']>0) & (df['close_rate']/df['closePrice']>=0.05),'pre_fast_up'] = 1
    80.     df['fast_up'] = 0
    81.     df.loc[(df['pre_fast_up']==1) & (df['pre_fast_up'].shift(1)==1) & (df['pre_fast_up'].shift(2)==1),'fast_up'] = 1
    82.     # 黄金交叉
    83.     df['gold_x'] = 0
    84.     df.loc[(df['ma5'].shift(1)'ma20'].shift(1)) & (df['ma5']>df['ma20']) & (df['ma5'].shift(-1)>df['ma20'].shift(-1)) & (df['ma5'].shift(-2)>df['ma20'].shift(-2)),'gold_x'] = 1
    85.     # 死亡交叉
    86.     df['dead_x'] = 0
    87.     df.loc[(df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['ma5']'ma20']) & (df['ma5'].shift(-1)'ma20'].shift(-1)) & (df['ma5'].shift(-2)'ma20'].shift(-2)),'dead_x'] = 1
    88.  
    89.  
    90.     # 1) 买入和持仓原则：
    91.     # 买入=1 持股=2(区间） 卖出=3 减仓=4 空仓=5(区间） 无信号=0
    92.     df['signal'] = 0
    93.     df['signal_reason'] = ''
    94.     # A. 股价向上突破定性线，定性线上行，买入。
    95.     # 翻译：收盘价涨跌幅为正 & 前一日收盘价小于定性价格 & 收盘价大于定性价格 & 定性线连续三日涨跌幅为正
    96.     df.loc[(df['close_rate']>0) & (df['closePrice'].shift(1)'ma20'].shift(1)) & (df['closePrice']>df['ma20']) & (df['ma20_direction']==1),'signal'] = 1
    97.     df.loc[(df['close_rate']>0) & (df['closePrice'].shift(1)'ma20'].shift(1)) & (df['closePrice']>df['ma20']) & (df['ma20_direction']==1),'signal_reason'] = 'A'
    98.     # B.定量线上穿定性线形成黄金交叉，买入。
    99.     # 翻译：黄金交叉
    100.     df.loc[df['gold_x']==1,'signal'] = 1
    101.     df.loc[df['gold_x']==1,'signal_reason'] = 'B'
    102.     # C.股价下跌，遇定性线上行支撑止跌回升，买入。
    103.     # 翻译：连续三日收盘价涨跌幅为负 & 定性线连续三日涨跌幅为正 & 股价被定性线支撑 & 下一日收盘价正跌幅为正
    104.     df.loc[(df['closePrice_direction']==-1) & (df['ma20_direction']==1) & (df['close_ma20_distance']<=2) & (df['close_rate'].shift(-1)>0),'signal'] = 1
    105.     df.loc[(df['closePrice_direction']==-1) & (df['ma20_direction']==1) & (df['close_ma20_distance']<=2) & (df['close_rate'].shift(-1)>0),'signal_reason'] = 'C'
    106.     # D.定性线上行，股价在定性线上方向上突破定量线，买入。
    107.     # 翻译：定性线连续三日涨跌幅为正 & 收盘价大于定性线价格 & （前一日收盘价小于前一日定量价格 & 当日收盘价大于定量价格）
    108.     df.loc[(df['ma20_direction']==1) & (df['closePrice']>df['ma20']) & (df['closePrice'].shift(1)'ma5'].shift(1)) & (df['closePrice']>df['ma5']),'signal'] = 1
    109.     df.loc[(df['ma20_direction']==1) & (df['closePrice']>df['ma20']) & (df['closePrice'].shift(1)'ma5'].shift(1)) & (df['closePrice']>df['ma5']),'signal_reason'] = 'D'
    110.     # E.定量线下行，遇定性线上行支撑止跌后再度上升，买入。
    111.     # 翻译：连续三日定量线涨跌幅为负 & 连续三日定性线涨跌幅为正 & 定量线被定性线支撑 & 下一日定量线涨跌幅为正
    112.     df.loc[(df['ma5_direction']==-1) & (df['ma20_direction']==1) & (df['ma5_ma20_distance']<=2) & (df['ma5_rate'].shift(-1)>0),'signal'] = 1
    113.     df.loc[(df['ma5_direction']==-1) & (df['ma20_direction']==1) & (df['ma5_ma20_distance']<=2) & (df['ma5_rate'].shift(-1)>0),'signal_reason'] = 'E'
    114.     # F.股价、定量线、定性线多头排列，持股
    115.     # 翻译：收盘价大于定量价格大于定性价格
    116.     df.loc[(df['closePrice']>df['ma5']) & (df['ma5']>df['ma20']),'signal'] = 2
    117.     df.loc[(df['closePrice']>df['ma5']) & (df['ma5']>df['ma20']),'signal_reason'] = 'F'
    118.  
    119.     # 2） 卖出和空仓原则
    120.     # A.股价跌破定性线，定性线走平或已拐头下行，卖出
    121.     # 翻译：前一日收盘价大于定性价格 & 当日收盘价小于定性价格 & 定性价格涨跌幅小于等于0
    122.     df.loc[(df['closePrice'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma20']) & (df['ma20_rate']<=0),'signal'] = 3
    123.     df.loc[(df['closePrice'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma20']) & (df['ma20_rate']<=0),'signal_reason'] = 'A'
    124.     # B.多头排列期间，股价跌破定量线，减仓
    125.     # 翻译：前一日收盘价大于定量价格大于定性价格 & 当日收盘价小于定量价格
    126.     df.loc[(df['closePrice'].shift(1)>df['ma5'].shift(1)) & (df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma5']),'signal'] = 4
    127.     df.loc[(df['closePrice'].shift(1)>df['ma5'].shift(1)) & (df['ma5'].shift(1)>df['ma20'].shift(1)) & (df['closePrice']'ma5']),'signal_reason'] = 'B'
    128.     # C.股价急速飙升，远离定性线，减仓
    129.     # 翻译：股价飙升 & 收盘价远离定性价格
    130.     df.loc[(df['fast_up']==1) & (df['close_ma5_distance']>=50),'signal'] = 4
    131.     df.loc[(df['fast_up']==1) & (df['close_ma5_distance']>=50),'signal_reason'] = 'C'
    132.     # D.定性线下行，空仓
    133.     # 翻译：连续三日定性价格涨跌幅为负
    134.     df.loc[df['ma20_direction']==-1,'signal'] = 5
    135.     df.loc[df['ma20_direction']==-1,'signal_reason'] = 'D'
    136.  
    137.     title_str = '双均线组合信号'
    138.     whole_header = ['日期','收盘价','开盘价','最高价','最低价','ma5','ma20']
    139.     whole_df = df
    140.     whole_pd_header = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice','ma5','ma20']
    141.     line_pd_header = ['ma5','ma20']
    142.     line_color_list = [0,2]
    143.  
    144.     line_data = {
    145.         'title_str':title_str,
    146.         'whole_header':whole_header,
    147.         'whole_df':whole_df,
    148.         'whole_pd_header':whole_pd_header,
    149.         'line_pd_header':line_pd_header,
    150.         'line_color_list':line_color_list
    151.     }
    152.     return line_data

    运行优化后的策略，为方便验证优化结果，把数据的时间往前拉长到10年，执行

     2014-05-06这个时间点的黄金交叉就靠谱多了

    数据

    链接：https://pan.baidu.com/s/1HPkMsDDyXTEgffoAVIhbZw 
    提取码：h80x 

  • 相关阅读:
    web前端设计与开发期末作品 用DIV CSS技术设计的凤阳旅游网站（web前端网页制作课作业）html css javascript
    Xshell工具连接本地虚拟机Linux系统
    leetcode(力扣) 221. 最大正方形（动态规划）
    操作系统：操作系统相关概念博客系统整理
    centos密码过期导致navicat无法通过SSH登录阿里云RDS问题
    iPaaS中API接口管理平台的作用
    基于C#的壁纸管理器（插件版） - 开源研究系列文章 - 个人小作品
    解决QT中文乱码
    企业使用微信管理系统更安全更高效更智能
    IIS 部署.NetCore，最细步骤
  • 原文地址：https://blog.csdn.net/m0_37967652/article/details/127614450