• PyQt5_寻找顶(底)背离并可视化

            技术指标的背离是指技术指标曲线的波动方向与价格曲线的趋势方向不一致，是使用技术指标最为重要的一点。在股市中，常见的技术指标的背离分为两种常见的形式，即顶背离和底背离。背离是预示市场走势即将见顶或者见底的依据，在价格还没有发生变化之前，技术指标会提前显示未来股价发展的趋势。

            顶部背离，是指在股市的上涨其中，股价的高点比前一点要高，而指标的高点却在下降，此时股票上涨的行情不会持久，投资者需谨慎，应考虑做空操作，以免被套牢。

            底部背离，是指股价下跌的过程中，技术指标虽然显示出比上一次的有所提高，但是股价仍然在下跌不止，此时股价的下跌行情快要见底。这是典型的买进信号，投资者可以做好准备，在适当的时间买入股票。

            在股市中，几乎所有的技术指标都有背离提示作用的功能，例如MACD、RSI和BOLL等。投资者可以用这些指标的背离功能来预测头部的风险和底部的买入机会，但在选择的时间参数上应当适当延长。由于日线存在着较多的骗线，一些技术指标会反复发出背离信号，使得其实用性不强，建议重点关注周线上的技术指标背离现象。

    PS:以上知识来自书籍《同花顺炒股从入门到精通》

    目录

    效果

    寻找背离策略代码

    可视化部分代码

    使用

    效果

    寻找背离策略代码

    思路：

    1. 找出K线的波峰波谷

    2. 找出指标的波峰波谷

    3. 将K线的波峰波谷和指标的波峰波谷对齐（就是将指标的高点和K线的高点对齐，指标的低点和K线的低点对齐，无法对齐的舍弃），本策略是K线高点和低点位置前后3个位置内指标出现了对应的高点和低点，就算是对齐，取最相近的那个点

    4. K线和指标的 高低点对齐后，每两个高点对比，如果K线升高，指标降低，或K线降低，指标升高，说明这对高点是顶背离；每两个低点对比，如果K线升高，指标降低，或K线降低，指标升高，说明是底背离。

    1. def res_peak_trough_list(df,peak_col,trough_col):
    2.     '''
    3.     寻找波峰波谷
    4.     :param df: 必须要有i_row
    5.     :param peak_col: 波峰字段
    6.     :param trough_col: 波谷字段
    7.     :return:
    8.     '''
    9.     # 循环寻找波峰和波谷,交替寻找
    10.     peak_list = []
    11.     trough_list = []
    12.     start_up_i_row = 0
    13.     peak_yeah = False
    14.     trough_yeah = False
    15.     while True:
    16.         if start_up_i_row >= len(df):
    17.             break
    18.         df00 = df.loc[df['i_row'] >= start_up_i_row].copy()
    19.         df00['peak'] = df00[peak_col].cummax()
    20.         df00['ext_00'] = df00['peak'] - df00['peak'].shift(-1)
    21.         df00_p = df00.loc[(df00['ext_00'] == 0) & (df00['ext_00'].shift(-1) == 0) & (df00['ext_00'].shift(-2) == 0) & (
    22.                     df00['ext_00'].shift(-3) == 0) & (df00['ext_00'].shift(-4) == 0)].copy()
    23.  
    24.         df00['trough'] = df00[trough_col].cummin()
    25.         df00['ext_10'] = df00['trough'] - df00['trough'].shift(-1)
    26.         df00_t = df00.loc[(df00['ext_10'] == 0) & (df00['ext_10'].shift(-1) == 0) & (df00['ext_10'].shift(-2) == 0) & (
    27.                     df00['ext_10'].shift(-3) == 0) & (df00['ext_10'].shift(-4) == 0)].copy()
    28.  
    29.         if (df00_p is None or len(df00_p) <= 0) and (df00_t is not None and len(df00_t) > 0):
    30.             trough_0 = df00_t.iloc[0]['i_row']
    31.             if trough_yeah:
    32.                 trough_list.append(trough_0)
    33.             break
    34.         elif (df00_t is None or len(df00_t) <= 0) and (df00_p is not None and len(df00_p) > 0):
    35.             peak_0 = df00_p.iloc[0]['i_row']
    36.             if peak_yeah:
    37.                 peak_list.append(peak_0)
    38.             break
    39.         elif (df00_p is None or len(df00_p) <= 0) or (df00_t is None or len(df00_t) <= 0):
    40.             break
    41.         else:
    42.             trough_0 = df00_t.iloc[0]['i_row']
    43.             peak_0 = df00_p.iloc[0]['i_row']
    44.  
    45.             if peak_0 == trough_0:
    46.                 start_up_i_row += 1
    47.                 continue
    48.             if peak_0 == 0:
    49.                 peak_yeah = False
    50.                 trough_yeah = True
    51.                 peak_list.append(peak_0)
    52.                 start_up_i_row = trough_0
    53.             elif trough_0 == 0:
    54.                 trough_yeah = False
    55.                 peak_yeah = True
    56.                 trough_list.append(trough_0)
    57.                 start_up_i_row = peak_0
    58.             else:
    59.                 if not trough_yeah and not peak_yeah:
    60.                     # 首次进入
    61.                     if peak_0 < trough_0:
    62.                         trough_yeah = True
    63.                         peak_list.append(peak_0)
    64.                         start_up_i_row = trough_0
    65.                     else:
    66.                         peak_yeah = True
    67.                         trough_list.append(trough_0)
    68.                         start_up_i_row = peak_0
    69.                 else:
    70.                     if peak_0 < trough_0:
    71.                         if peak_yeah:
    72.                             peak_list.append(peak_0)
    73.                             peak_yeah = False
    74.                             trough_yeah = True
    75.                         start_up_i_row = trough_0
    76.                     else:
    77.                         if trough_yeah:
    78.                             trough_list.append(trough_0)
    79.                             trough_yeah = False
    80.                             peak_yeah = True
    81.                         start_up_i_row = peak_0
    82.         pass
    83.  
    84.     two_list = peak_list + trough_list
    85.     two_list.sort()
    86.     i = 0
    87.     remove_list = []
    88.     while i < len(two_list) - 2:
    89.         one_i = two_list[i]
    90.         two_i = two_list[i + 1]
    91.         if one_i in peak_list and two_i in trough_list:
    92.             if abs(one_i - two_i) <= 2:
    93.                 i += 2
    94.                 remove_list.append(one_i)
    95.                 remove_list.append(two_i)
    96.                 continue
    97.             pass
    98.         elif one_i in trough_list and two_i in peak_list:
    99.             if abs(one_i - two_i) <= 2:
    100.                 i += 2
    101.                 remove_list.append(one_i)
    102.                 remove_list.append(two_i)
    103.                 continue
    104.             pass
    105.         else:
    106.             pass
    107.         i += 1
    108.         pass
    109.  
    110.     two_list00 = []
    111.     for item in two_list:
    112.         if item in remove_list:
    113.             continue
    114.         two_list00.append(item)
    115.     peak_list00 = []
    116.     trough_list00 = []
    117.     for item in two_list00:
    118.         if item in peak_list:
    119.             peak_list00.append(item)
    120.         if item in trough_list:
    121.             trough_list00.append(item)
    122.     return peak_list00,trough_list00
    123.  
    124. # 顶背离 和 底背离 索引对齐步骤
    125. def res_alignment_top_and_bottom_deviation_index(first_list,second_list):
    126.     '''
    127.     顶背离 和 底背离
    128.     1. 对比时前后参差超过3，跳过
    129.     :param first_list: 升序排列
    130.     :param second_list: 升序排列
    131.     :return:
    132.     '''
    133.     first_list00 = []
    134.     second_list00 = []
    135.     i = 0
    136.     j = 0
    137.     while i<len(first_list) and j<len(second_list):
    138.         i_node = first_list[i]
    139.         j0_node_list = []
    140.         for j0 in range(j,len(second_list)):
    141.             j0_node_list.append(abs(i_node-second_list[j0]))
    142.             pass
    143.         min_j0 = min(j0_node_list)
    144.         if min_j0>3:
    145.             i += 1
    146.             continue
    147.         min_index = j0_node_list.index(min_j0)
    148.         ori_index = j + min_index
    149.         first_list00.append(i_node)
    150.         second_list00.append(second_list[ori_index])
    151.         i += 1
    152.         j = ori_index + 1
    153.  
    154.     return first_list00,second_list00
    155.  
    156. def res_top_and_bottom_deviation(df,k_col_name_list,f_col_name_list):
    157.     '''
    158.     顶背离 和 底背离
    159.     :param df: 必要要有 i_row 列
    160.     :param k_col_name_list: K线的[高点列，低点列]
    161.     :param f_col_name_list: 指标的[高点列，低点列]
    162.     :return:
    163.     '''
    164.     k_peak, k_trough = res_peak_trough_list(df, k_col_name_list[0], k_col_name_list[1])
    165.     f_peak, f_trought = res_peak_trough_list(df, f_col_name_list[0], f_col_name_list[1])
    166.     k_top, f_top = res_alignment_top_and_bottom_deviation_index(k_peak, f_peak)
    167.     k_bottom, f_bottom = res_alignment_top_and_bottom_deviation_index(k_trough, f_trought)
    168.  
    169.     k_top_list = []
    170.     f_top_list = []
    171.     i = 0
    172.     while i < len(k_top) - 2:
    173.         one_k = k_top[i]
    174.         two_k = k_top[i + 1]
    175.         one_f = f_top[i]
    176.         two_f = f_top[i + 1]
    177.         one_k_val = df.iloc[one_k][k_col_name_list[0]]
    178.         two_k_val = df.iloc[two_k][k_col_name_list[0]]
    179.         one_f_val = df.iloc[one_f][f_col_name_list[0]]
    180.         two_f_val = df.iloc[two_f][f_col_name_list[0]]
    181.         if (one_k_val > two_k_val and one_f_val < two_f_val) or (one_k_val < two_k_val and one_f_val > two_f_val):
    182.             k_top_list.append([(one_k, one_k_val), (two_k, two_k_val)])
    183.             f_top_list.append([(one_f, one_f_val), (two_f, two_f_val)])
    184.             i += 2
    185.             pass
    186.         else:
    187.             i += 1
    188.         pass
    189.  
    190.     k_bottom_list = []
    191.     f_bottom_list = []
    192.     i = 0
    193.     while i < len(k_bottom) - 2:
    194.         one_k = k_bottom[i]
    195.         two_k = k_bottom[i + 1]
    196.         one_f = f_bottom[i]
    197.         two_f = f_bottom[i + 1]
    198.         one_k_val = df.iloc[one_k][k_col_name_list[1]]
    199.         two_k_val = df.iloc[two_k][k_col_name_list[1]]
    200.         one_f_val = df.iloc[one_f][f_col_name_list[1]]
    201.         two_f_val = df.iloc[two_f][f_col_name_list[1]]
    202.         if (one_k_val > two_k_val and one_f_val < two_f_val) or (one_k_val < two_k_val and one_f_val > two_f_val):
    203.             k_bottom_list.append([(one_k, one_k_val), (two_k, two_k_val)])
    204.             f_bottom_list.append([(one_f, one_f_val), (two_f, two_f_val)])
    205.             i += 2
    206.             pass
    207.         else:
    208.             i += 1
    209.         pass
    210.  
    211.     return k_top_list,f_top_list,k_bottom_list,f_bottom_list

    可视化部分代码

    需要导入的包、日期横坐标控件、K线图控件的代码请查看本人本栏目中其他博文，这里不再赘述。

    本文以KDJ指标为例，KDJ显示控件

    1. class KDJ_PlotWidget(pg.PlotWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.setLabel('left', 'KDJ')
    5.         xax = RotateAxisItem(orientation='bottom')
    6.         xax.setHeight(h=60)
    7.         self.setAxisItems({'bottom':xax})
    8.  
    9.         self.color_k = (255,255,0) # 纯黄
    10.         self.color_d = (30,144,255) # 道奇蓝
    11.         self.color_j = (255,0,255) # 紫红色
    12.  
    13.         self.color_white = (248,248,255)
    14.         pass
    15.     def set_data(self,data:Dict[str,Any]):
    16.         df = data['df']
    17.         df.reset_index(inplace=True)
    18.         tradeDate_list = df['tradeDate'].values.tolist()
    19.         x = range(len(df))
    20.         xTick_show = []
    21.         x_dur = math.ceil(len(df)/20)
    22.         for i in range(0,len(df),x_dur):
    23.             xTick_show.append((i,tradeDate_list[i]))
    24.         if len(df)%20!=0:
    25.             xTick_show.append((len(df)-1,tradeDate_list[-1]))
    26.  
    27.         self.clear()
    28.         self.addLegend()
    29.         xax = self.getAxis('bottom')
    30.         xax.setTicks([xTick_show])
    31.  
    32.         kdjk_fixed_target = pg.PlotCurveItem(x=np.array(x),y=np.array(df['kdj_k'].values.tolist()),name='K',pen=pg.mkPen({'color':self.color_k,'width':1}),connect='finite')
    33.         kdjd_fixed_target = pg.PlotCurveItem(x=np.array(x),y=np.array(df['kdj_d'].values.tolist()),name='D',pen=pg.mkPen({'color':self.color_d,'width':1}),connect='finite')
    34.         kdjj_fixed_target = pg.PlotCurveItem(x=np.array(x),y=np.array(df['kdj_j'].values.tolist()),name='J',pen=pg.mkPen({'color':self.color_j,'width':1}),connect='finite')
    35.  
    36.         self.addItem(kdjk_fixed_target)
    37.         self.addItem(kdjd_fixed_target)
    38.         self.addItem(kdjj_fixed_target)
    39.  
    40.         line_one_target = pg.InfiniteLine(pos=(0, 0), movable=False, angle=0,
    41.                                            pen=pg.mkPen({'color': self.color_white, 'width': 1}),
    42.                                           label='0',
    43.                                           labelOpts=({'position':0.05})
    44.                                            )
    45.         line_two_target = pg.InfiniteLine(pos=(0, 20), movable=False, angle=0,
    46.                                          pen=pg.mkPen({'color': self.color_white, 'width': 1}),
    47.                                           label='20',
    48.                                           labelOpts=({'position': 0.05})
    49.                                          )
    50.  
    51.         line_three_target = pg.InfiniteLine(pos=(0, 50), movable=False, angle=0,
    52.                                           pen=pg.mkPen({'color': self.color_white, 'width': 1}),
    53.                                             label='50',
    54.                                             labelOpts=({'position': 0.05})
    55.                                           )
    56.         line_four_target = pg.InfiniteLine(pos=(0, 80), movable=False, angle=0,
    57.                                             pen=pg.mkPen({'color': self.color_white, 'width': 1}),
    58.                                             label='80',
    59.                                            labelOpts=({'position': 0.05})
    60.                                             )
    61.         self.addItem(line_one_target)
    62.         self.addItem(line_two_target)
    63.         self.addItem(line_three_target)
    64.         self.addItem(line_four_target)
    65.  
    66.         self.enableAutoRange()
    67.         pass

    结果显示控件

    1. class PyQtGraphScrollKWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.factor_widgets = {}
    5.         self.init_data()
    6.         self.init_ui()
    7.         pass
    8.     def init_data(self):
    9.         # https://www.sioe.cn/yingyong/yanse-rgb-16/
    10.         # self.color_line = (30, 144, 255)
    11.         self.color_line = (255, 255, 0)
    12.         self.color_highligh = (220,20,60)
    13.         # 0 幽灵的白色; 1 纯黄; 2 紫红色; 3 纯绿; 4 道奇蓝
    14.         self.color_list = [(248, 248, 255), (255, 255, 0), (255, 0, 255), (0, 128, 0), (30, 144, 255)]
    15.         self.color_green = (34,139,34)
    16.         self.color_red = (220,20,60)
    17.         self.main_fixed_target_list = []  # 主体固定曲线，不能被删除
    18.         self.whole_df = None
    19.         self.whole_header = None
    20.         self.whole_pd_header = None
    21.         self.current_whole_data = None
    22.         self.current_whole_df = None
    23.         self.factor_list = None
    24.         self.ma_list = None
    25.         self.show_duration = []
    26.         self.show_line = []
    27.         self.show_segment = []
    28.         self.detail_map = None
    29.  
    30.         self.factor_code_widgetname_map = {
    31.             'KDJ':'KDJ_PlotWidget'
    32.         }
    33.         pass
    34.     def init_ui(self):
    35.         self.whole_duration_label = QtWidgets.QLabel('左边界~右边界')
    36.         pic_download_btn = QtWidgets.QPushButton('滚动截图')
    37.         pic_download_btn.clicked.connect(self.pic_download_btn_clicked)
    38.         layout_top = QtWidgets.QHBoxLayout()
    39.         layout_top.addWidget(self.whole_duration_label)
    40.         layout_top.addStretch(1)
    41.         layout_top.addWidget(pic_download_btn)
    42.  
    43.         self.title_label = QtWidgets.QLabel('执行过程查看')
    44.         self.title_label.setAlignment(Qt.AlignCenter)
    45.         self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')
    46.  
    47.         # 滚动区域开始
    48.         self.pw_layout = QtWidgets.QVBoxLayout()
    49.         self.scroll_area = QtWidgets.QScrollArea()
    50.         self.scroll_area.setWidgetResizable(True)
    51.         # self.scroll_area.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
    52.  
    53.         layout_right = QtWidgets.QVBoxLayout()
    54.         layout_right.addWidget(self.title_label)
    55.         layout_right.addLayout(layout_top)
    56.         layout_right.addWidget(self.scroll_area)
    57.         self.setLayout(layout_right)
    58.         pass
    59.  
    60.     def set_data(self, data: Dict[str, Any]):
    61.         title_str = data['title_str']
    62.         whole_header = data['whole_header']
    63.         whole_df = data['whole_df']
    64.         whole_pd_header = data['whole_pd_header']
    65.         factor_list = data['factor_list']
    66.         ma_list = data['ma_list']
    67.         detail_map = data['detail_map']
    68.  
    69.  
    70.         self.whole_header = whole_header
    71.         self.whole_df = whole_df
    72.         self.whole_pd_header = whole_pd_header
    73.         self.factor_list = factor_list
    74.         self.ma_list = ma_list
    75.         if detail_map.get('duration') is not None:
    76.             self.show_duration = detail_map['duration']
    77.         if detail_map.get('line') is not None:
    78.             self.show_line = detail_map['line']
    79.         if detail_map.get('segment') is not None:
    80.             self.show_segment = detail_map['segment']
    81.         self.detail_map = detail_map
    82.  
    83.  
    84.         self.title_label.setText(title_str)
    85.         self.whole_duration_label.setText(f"{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")
    86.  
    87.         self.current_whole_df = self.whole_df.copy()
    88.         self.caculate_and_show_data()
    89.         pass
    90.  
    91.     def caculate_and_show_data(self):
    92.         df = self.current_whole_df.copy()
    93.         # df.reset_index(inplace=True)
    94.         df['i_count'] = [i for i in range(len(df))]
    95.         tradeDate_list = df['tradeDate'].values.tolist()
    96.         x = range(len(df))
    97.         xTick_show = []
    98.         x_dur = math.ceil(len(df) / 20)
    99.         for i in range(0, len(df), x_dur):
    100.             xTick_show.append((i, tradeDate_list[i]))
    101.         if len(df) % 20 != 0:
    102.             xTick_show.append((len(df) - 1, tradeDate_list[-1]))
    103.         candle_data = []
    104.         for i, row in df.iterrows():
    105.             candle_data.append(
    106.                 (row['i_count'], row['openPrice'], row['closePrice'], row['lowestPrice'], row['highestPrice']))
    107.  
    108.         self.current_whole_data = df.loc[:, self.whole_pd_header].values.tolist()
    109.         # 开始配置显示的内容
    110.         self.create_candle_widget()
    111.         self.factor_widgets.clear()
    112.  
    113.         xax = self.pw.getAxis('bottom')
    114.         xax.setTicks([xTick_show])
    115.  
    116.         # 标记技术图形 start
    117.         if len(self.show_duration)>0:
    118.             duration_list = self.show_duration[0]
    119.             duration_color = self.show_duration[1]
    120.             for i,item in enumerate(duration_list):
    121.                 for item00 in item:
    122.                     signal_fiexed_target = pg.LinearRegionItem([item00[0], item00[1]],
    123.                                                                movable=False, brush=(
    124.                             self.color_list[duration_color[i]][0], self.color_list[duration_color[i]][1], self.color_list[duration_color[i]][2], 50))
    125.                     self.pw.addItem(signal_fiexed_target)
    126.                 pass
    127.             pass
    128.         if len(self.show_line)>0:
    129.             line_list = self.show_line[0]
    130.             line_color = self.show_line[1]
    131.             for i,item in enumerate(line_list):
    132.                 for item00 in item:
    133.                     signal_fiexed_target = pg.InfiniteLine(pos=(item00, 0), movable=False, angle=90,
    134.                                                            pen=pg.mkPen({'color': self.color_list[line_color[i]], 'width': 1})
    135.                                                            )
    136.                     self.pw.addItem(signal_fiexed_target)
    137.             pass
    138.         if len(self.show_segment)>0:
    139.             for item in self.show_segment:
    140.                 r_target = pg.LineSegmentROI(item,pen={'color':self.color_red,'width':2},movable=False)
    141.                 self.pw.addItem(r_target)
    142.             pass
    143.         # 标记技术图形 end
    144.  
    145.         candle_fixed_target = CandlestickItem(candle_data)
    146.         self.main_fixed_target_list.append(candle_fixed_target)
    147.         self.pw.addItem(candle_fixed_target)
    148.  
    149.         if len(self.ma_list)>0:
    150.             for i, item in enumerate(self.ma_list):
    151.                 item_color = i%len(self.color_list)
    152.                 line_fixed_target = pg.PlotCurveItem(x=np.array(x), y=np.array(df[item].values.tolist()),
    153.                                                      pen=pg.mkPen({'color': self.color_list[item_color],
    154.                                                                    'width': 1}),
    155.                                                      connect='finite')
    156.                 self.main_fixed_target_list.append(line_fixed_target)
    157.                 self.pw.addItem(line_fixed_target)
    158.             pass
    159.  
    160.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    161.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    162.         self.label = pg.TextItem()
    163.  
    164.         self.pw.addItem(self.vLine, ignoreBounds=True)
    165.         self.pw.addItem(self.hLine, ignoreBounds=True)
    166.         self.pw.addItem(self.label, ignoreBounds=True)
    167.  
    168.         self.vb = self.pw.getViewBox()
    169.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    170.  
    171.         # 其他项
    172.         for item in self.factor_list:
    173.             item_widget = eval(self.factor_code_widgetname_map[item])()
    174.             item_widget.setMinimumHeight(300)
    175.             item_widget.set_data({'df':self.current_whole_df.copy()})
    176.             item_widget.setXLink(self.pw)
    177.             # 标记 start
    178.             if self.detail_map.get(item) is not None:
    179.                 target_detail = self.detail_map[item]
    180.                 if target_detail.get('duration') is not None:
    181.                     target_duration = target_detail['duration']
    182.                     target_list = target_duration[0]
    183.                     target_color = target_duration[1]
    184.                     for i,i0 in enumerate(target_list):
    185.                         for i00 in i0:
    186.                             signal_fiexed_target = pg.LinearRegionItem([i00[0], i00[1]],
    187.                                                                        movable=False, brush=(
    188.                                     self.color_list[target_color[i]][0], self.color_list[target_color[i]][1],
    189.                                     self.color_list[target_color[i]][2], 50))
    190.                             item_widget.addItem(signal_fiexed_target)
    191.                     pass
    192.                 if target_detail.get('line') is not None:
    193.                     target_line = target_detail['line']
    194.                     target_list = target_line[0]
    195.                     target_color = target_line[1]
    196.                     for i, i0 in enumerate(target_list):
    197.                         for i00 in i0:
    198.                             signal_fiexed_target = pg.InfiniteLine(pos=(i00, 0), movable=False, angle=90,
    199.                                                                    pen=pg.mkPen(
    200.                                                                        {'color': self.color_list[target_color[i]],
    201.                                                                         'width': 1})
    202.                                                                    )
    203.                             item_widget.addItem(signal_fiexed_target)
    204.                     pass
    205.                 if target_detail.get('segment') is not None:
    206.                     target_segment = target_detail['segment']
    207.                     for i00 in target_segment:
    208.                         r_target = pg.LineSegmentROI(i00, pen={'color': self.color_red, 'width': 2}, movable=False)
    209.                         item_widget.addItem(r_target)
    210.                     pass
    211.             # 标记 end
    212.             self.factor_widgets[item] = item_widget
    213.         self.fill_pw_widget()
    214.         self.pw.enableAutoRange()
    215.         pass
    216.  
    217.     def mouseMoved(self, evt):
    218.         pos = evt[0]
    219.         if self.pw.sceneBoundingRect().contains(pos):
    220.             mousePoint = self.vb.mapSceneToView(pos)
    221.             index = int(mousePoint.x())
    222.             if index >= 0 and index < len(self.current_whole_data):
    223.                 target_data = self.current_whole_data[index]
    224.                 html_str = ''
    225.                 for i, item in enumerate(self.whole_header):
    226.                     html_str += f"
      {item}:{target_data[i]}"
    227.                 self.label.setHtml(html_str)
    228.                 self.label.setPos(mousePoint.x(), mousePoint.y())
    229.             self.vLine.setPos(mousePoint.x())
    230.             self.hLine.setPos(mousePoint.y())
    231.         pass
    232.  
    233.     def mouseClicked(self, evt):
    234.         pass
    235.  
    236.     def updateViews(self):
    237.         pass
    238.  
    239.     def fill_pw_widget(self):
    240.         # 清空控件
    241.         while self.pw_layout.count():
    242.             item = self.pw_layout.takeAt(0)
    243.             widget = item.widget()
    244.             if widget is not None:
    245.                 widget.deleteLater()
    246.                 pass
    247.             pass
    248.         sc_child_widget = self.scroll_area.takeWidget()
    249.         if sc_child_widget is not None:
    250.             sc_child_widget.deleteLater()
    251.         self.pw_layout.addWidget(self.pw)
    252.         for item in self.factor_widgets.values():
    253.             self.pw_layout.addWidget(item)
    254.         one_sc_child_widget = QtWidgets.QWidget()
    255.         one_sc_child_widget.setLayout(self.pw_layout)
    256.         self.scroll_area.setWidget(one_sc_child_widget)
    257.         pass
    258.  
    259.     def create_candle_widget(self):
    260.         xax = RotateAxisItem(orientation='bottom')
    261.         xax.setHeight(h=60)
    262.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    263.         self.pw.setMinimumHeight(500)
    264.         self.pw.setMouseEnabled(x=True, y=True)
    265.         # self.pw.enableAutoRange(x=False,y=True)
    266.         self.pw.setAutoVisible(x=False, y=True)
    267.         pass
    268.  
    269.     def pic_download_btn_clicked(self):
    270.         now_str = datetime.now().strftime('%Y%m%d%H%M%S')
    271.         path,_ = QtWidgets.QFileDialog.getSaveFileName(
    272.             self,
    273.             '选择图片保存路径',
    274.             f"pic_{now_str}",
    275.             'JPG(*.jpg)'
    276.         )
    277.         if not path:
    278.             return
    279.  
    280.         widget = self.scroll_area.widget()
    281.         pix = widget.grab()
    282.         pix.save(path)
    283.         pass
    284.  
    285.     pass

    使用

    1. if __name__ == '__main__':
    2.     pre_path = r'D:/temp005/600660.csv'
    3.     caculate_start_date_str = '2020-01-01'
    4.     df = pd.read_csv(pre_path,encoding='utf-8')
    5.     # 删除停牌的数据
    6.     df = df.loc[df['openPrice'] > 0].copy()
    7.     df['o_date'] = df['tradeDate']
    8.     df['o_date'] = pd.to_datetime(df['o_date'])
    9.     df = df.loc[df['o_date'] >= caculate_start_date_str].copy()
    10.     # 保存未复权收盘价数据
    11.     df['close'] = df['closePrice']
    12.     # 计算前复权数据
    13.     df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    14.     df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    15.     df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    16.     df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    17.  
    18.     df.reset_index(inplace=True)
    19.     df['i_row'] = [i for i in range(len(df))]
    20.  
    21.     factor_list = ['KDJ']
    22.     df['kdj_k'], df['kdj_d'] = talib.STOCH(df['highestPrice'], df['lowestPrice'], df['closePrice'], fastk_period=5,
    23.                                            slowk_period=3, slowk_matype=0, slowd_period=3, slowd_matype=0)
    24.     df['kdj_j'] = 3 * df['kdj_k'] - 2 * df['kdj_d']
    25.  
    26.     k_top_list, f_top_list, k_bottom_list, f_bottom_list = res_top_and_bottom_deviation(df,['highestPrice','lowestPrice'],['kdj_k','kdj_k'])
    27.  
    28.     k_deviation_list = k_top_list + k_bottom_list
    29.     f_deviation_list = f_top_list + f_bottom_list
    30.     detail_map = {
    31.         'segment': k_deviation_list,
    32.         'KDJ': {
    33.             'segment': f_deviation_list
    34.         }
    35.     }
    36.  
    37.     columns_list = ['日期', '收盘价', '开盘价', '最高价', '最低价']
    38.     columns_pd_list = ['tradeDate', 'closePrice', 'openPrice', 'highestPrice', 'lowestPrice']
    39.     line_data = {
    40.         'title_str': '寻找顶背离和底背离',
    41.         'whole_header': columns_list,
    42.         'whole_df': df,
    43.         'whole_pd_header': columns_pd_list,
    44.         'detail_map': detail_map,
    45.         'factor_list': factor_list,
    46.         'ma_list': [],
    47.         'peak_trough_data': None
    48.     }
    49.  
    50.     QtCore.QCoreApplication.setAttribute(QtCore.Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    51.     app = QtWidgets.QApplication(sys.argv)
    52.     t_win = PyQtGraphScrollKWidget()
    53.     t_win.set_data(line_data)
    54.     t_win.showMaximized()
    55.     app.exec()
    56.     pass
  • 相关阅读:
    QGIS编译（跨平台编译）之五十一：qgis_native库在Qt Creator环境下编译的错误处理
    系统设计的端到端原则
    死锁产生的条件及其预防
    再见print()，有了 PySnooper 可轻松 debug 问题所在
    虹科直播 | CDS网络与数据安全专题技术直播重磅来袭，11.2起与您精彩相约
    无法直接打印变量值，是哪一步出错了吗？
    Django+vue前后端分离实战--vue后台管理系统--vue环境安装项目创建
    Linux运维相关基础知识
    InnoDB 数据页结构
    EN 14351-1门窗及配件—CE认证
  • 原文地址：https://blog.csdn.net/m0_37967652/article/details/128122115