效果

代码

需要的包、蜡烛控件、日期横轴控件、一些日期计算方法

import sys,json
import numpy as np
from datetime import datetime
from dateutil.relativedelta import relativedelta
from typing import Dict,Any
from PyQt5 import QtCore,QtGui,QtWidgets
from PyQt5.QtCore import Qt
import pyqtgraph as pg
import pyqtgraph.examples
pg.setConfigOption('background','k')
pg.setConfigOption('foreground','w')# 返回今天的日期
def res_date_normal_str():return datetime.now().strftime('%Y-%m-%d')
# 返回今年第一天
def res_current_year_first_day():now_day = datetime.today()current_year = now_day.yearres_str = f"{current_year}-01-01"return res_strpass
# 往回推一年的第一天
def res_pre_year_first_day():# https://blog.csdn.net/weixin_42185136/article/details/108646120pre_year_day = (datetime.now()-relativedelta(years=1)).strftime('%Y-%m-%d')return pre_year_day
# 往回推两年的第一天
def res_pre_two_year_first_day():pre_year_day = (datetime.now() - relativedelta(years=2)).strftime('%Y-%m-%d')return pre_year_dayclass RotateAxisItem(pg.AxisItem):def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):p.setRenderHint(p.Antialiasing,False)p.setRenderHint(p.TextAntialiasing,True)## draw long line along axispen,p1,p2 = axisSpecp.setPen(pen)p.drawLine(p1,p2)p.translate(0.5,0)  ## resolves some damn pixel ambiguity## draw ticksfor pen,p1,p2 in tickSpecs:p.setPen(pen)p.drawLine(p1,p2)## draw all text# if self.tickFont is not None:#     p.setFont(self.tickFont)p.setPen(self.pen())for rect,flags,text in textSpecs:# this is the important partp.save()p.translate(rect.x(),rect.y())p.rotate(-30)p.drawText(-rect.width(),rect.height(),rect.width(),rect.height(),flags,text)# restoring the painter is *required*!!!p.restore()class CandlestickItem(pg.GraphicsObject):def __init__(self, data):pg.GraphicsObject.__init__(self)self.data = data  ## data must have fields: time, open, close, min, maxself.generatePicture()def generatePicture(self):## pre-computing a QPicture object allows paint() to run much more quickly,## rather than re-drawing the shapes every time.self.picture = QtGui.QPicture()p = QtGui.QPainter(self.picture)p.setPen(pg.mkPen('d'))w = (self.data[1][0] - self.data[0][0]) / 3.for (t, open, close, min, max) in self.data:p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))if open > close:p.setBrush(pg.mkBrush('r'))else:p.setBrush(pg.mkBrush('g'))p.drawRect(QtCore.QRectF(t - w, open, w * 2, close - open))p.end()def paint(self, p, *args):p.drawPicture(0, 0, self.picture)def boundingRect(self):## boundingRect _must_ indicate the entire area that will be drawn on## or else we will get artifacts and possibly crashing.## (in this case, QPicture does all the work of computing the bouning rect for us)return QtCore.QRectF(self.picture.boundingRect())

K线图与指标同步缩放并排显示控件

class PyQtGraphLineWidget(QtWidgets.QWidget):def __init__(self):super().__init__()self.init_data()self.init_ui()passdef init_data(self):self.please_select_str = '---请选择---'self.func_map = {}self.func_item_list = []self.duration_map = {'今年':'a','最近一年':'b','最近两年':'c'}# https://www.sioe.cn/yingyong/yanse-rgb-16/self.color_line = (30, 144, 255)self.color_ma_5 = (248,248,255) # 幽灵的白色self.color_ma_10 = (255,255,0) # 纯黄self.color_ma_20 = (255,0,255) # 紫红色self.color_ma_30 = (0,128,0) # 纯绿self.color_ma_60 = (30,144,255) # 道奇蓝self.color_up = (220,20,60)self.color_down = (60,179,113)self.main_fixed_target_list = []  # 主体固定曲线，不能被删除self.whole_df = Noneself.whole_header = Noneself.whole_header2 = Noneself.whole_pd_header = Noneself.whole_pd_header2 = Noneself.current_whole_data = Noneself.current_whole_data2 = Noneself.current_whole_df = Nonepassdef init_ui(self):# 控制面板 startleft_tip = QtWidgets.QLabel('左边界')self.left_point = QtWidgets.QDateEdit()self.left_point.setDisplayFormat('yyyy-MM-dd')self.left_point.setCalendarPopup(True)right_tip = QtWidgets.QLabel('右边界')self.right_point = QtWidgets.QDateEdit()self.right_point.setDisplayFormat('yyyy-MM-dd')self.right_point.setCalendarPopup(True)duration_sel_btn = QtWidgets.QPushButton('确定')duration_sel_btn.clicked.connect(self.duration_sel_btn_clicked)duration_reset_btn = QtWidgets.QPushButton('重置')duration_reset_btn.clicked.connect(self.duration_reset_btn_clicked)duration_tip = QtWidgets.QLabel('常用时间')self.duration_combox = QtWidgets.QComboBox()self.duration_combox.addItem(self.please_select_str)self.duration_combox.addItems(list(self.duration_map.keys()))self.duration_combox.currentIndexChanged.connect(self.duration_combox_currentIndexChanged)combox_tip = QtWidgets.QLabel('功能')self.func_combox = QtWidgets.QComboBox()self.func_combox.addItem(self.please_select_str)self.func_combox.addItems(list(self.func_map.keys()))self.func_combox.currentIndexChanged.connect(self.func_combox_currentIndexChanged)clear_func_btn = QtWidgets.QPushButton('清空功能区')clear_func_btn.clicked.connect(self.clear_func_btn_clicked)self.whole_duration_label = QtWidgets.QLabel('原始最宽边界：左边界~右边界')self.now_duration_label = QtWidgets.QLabel('当前显示最宽边界：左边界~右边界')layout_date = QtWidgets.QHBoxLayout()layout_date.addWidget(left_tip)layout_date.addWidget(self.left_point)layout_date.addWidget(right_tip)layout_date.addWidget(self.right_point)layout_date.addWidget(duration_sel_btn)layout_date.addWidget(duration_reset_btn)layout_date.addSpacing(30)layout_date.addWidget(duration_tip)layout_date.addWidget(self.duration_combox)layout_date.addSpacing(30)layout_date.addWidget(combox_tip)layout_date.addWidget(self.func_combox)layout_date.addWidget(clear_func_btn)layout_date.addStretch(1)layout_duration = QtWidgets.QHBoxLayout()layout_duration.addWidget(self.whole_duration_label)layout_duration.addSpacing(30)layout_duration.addWidget(self.now_duration_label)layout_duration.addStretch(1)# 控制面板 endself.title_label = QtWidgets.QLabel('均线训练')self.title_label.setAlignment(Qt.AlignCenter)self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')xax = RotateAxisItem(orientation='bottom')xax.setHeight(h=60)xax2 = RotateAxisItem(orientation='bottom')xax2.setHeight(h=60)self.pw = pg.PlotWidget(axisItems={'bottom': xax})self.pw2 = pg.PlotWidget(axisItems={'bottom': xax2})# self.pw.setMouseEnabled(x=True, y=True)# self.pw.enableAutoRange(x=False,y=True)# self.pw.setAutoVisible(x=False, y=True)layout_pw = QtWidgets.QVBoxLayout()layout_pw.addWidget(self.pw,3)layout_pw.addWidget(self.pw2,1)layout = QtWidgets.QVBoxLayout()layout.addWidget(self.title_label)layout.addLayout(layout_date)layout.addLayout(layout_duration)layout.addLayout(layout_pw)self.setLayout(layout)passdef set_data(self,data:Dict[str,Any]):title_str = data['title_str']whole_header = data['whole_header']whole_header2 = data['whole_header2']whole_df = data['whole_df']whole_pd_header = data['whole_pd_header']whole_pd_header2 = data['whole_pd_header2']self.whole_header = whole_headerself.whole_header2 = whole_header2self.whole_df = whole_dfself.whole_pd_header = whole_pd_headerself.whole_pd_header2 = whole_pd_header2self.title_label.setText(title_str)self.whole_duration_label.setText(f"原始最宽边界：{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")self.current_whole_df = self.whole_df.copy()self.caculate_and_show_data()passdef caculate_and_show_data(self):df = self.current_whole_df.copy()df.reset_index(inplace=True)tradeDate_list = df['tradeDate'].values.tolist()x = range(len(df))xTick_show = []x_dur = math.ceil(len(df)/20)for i in range(0,len(df),x_dur):xTick_show.append((i,tradeDate_list[i]))if len(df)%20 != 0:xTick_show.append((len(df)-1,tradeDate_list[-1]))candle_data = []for i,row in df.iterrows():candle_data.append((i,row['openPrice'],row['closePrice'],row['lowestPrice'],row['highestPrice']))self.current_whole_data = df.loc[:,self.whole_pd_header].values.tolist()self.current_whole_data2 = df.loc[:,self.whole_pd_header2].values.tolist()# 开始配置显示的内容self.pw.clear()self.pw2.clear()self.func_item_list.clear()self.now_duration_label.setText(f"当前显示最宽边界：{df.iloc[0]['tradeDate']}~{df.iloc[-1]['tradeDate']}")xax = self.pw.getAxis('bottom')xax.setTicks([xTick_show])xax2 = self.pw2.getAxis('bottom')xax2.setTicks([xTick_show])candle_fixed_target = CandlestickItem(candle_data)self.main_fixed_target_list.append(candle_fixed_target)self.pw.addItem(candle_fixed_target)# 指标indicator_fixed_target = pg.PlotCurveItem(x=np.array(x), y=np.array(df['rsi'].values.tolist()),pen=pg.mkPen({'color': self.color_line, 'width': 2}),connect='finite')self.main_fixed_target_list.append(indicator_fixed_target)self.pw2.addItem(indicator_fixed_target)self.pw2.setXLink(self.pw)self.vLine = pg.InfiniteLine(angle=90, movable=False)self.hLine = pg.InfiniteLine(angle=0, movable=False)self.label = pg.TextItem()self.vLine2 = pg.InfiniteLine(angle=90, movable=False)self.hLine2 = pg.InfiniteLine(angle=0, movable=False)self.label2 = pg.TextItem()self.pw.addItem(self.vLine, ignoreBounds=True)self.pw.addItem(self.hLine, ignoreBounds=True)self.pw.addItem(self.label, ignoreBounds=True)self.pw2.addItem(self.vLine2, ignoreBounds=True)self.pw2.addItem(self.hLine2, ignoreBounds=True)self.pw2.addItem(self.label2, ignoreBounds=True)self.vb = self.pw.getViewBox()self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)self.pw.enableAutoRange()self.vb2 = self.pw2.getViewBox()self.proxy2 = pg.SignalProxy(self.pw2.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved2)self.pw2.enableAutoRange()passdef mouseMoved(self,evt):pos = evt[0]if self.pw.sceneBoundingRect().contains(pos):mousePoint = self.vb.mapSceneToView(pos)index = int(mousePoint.x())if index>=0 and index<len(self.current_whole_data):target_data = self.current_whole_data[index]html_str = ''for i,item in enumerate(self.whole_header):html_str += f"<br/>{item}:{target_data[i]}"self.label.setHtml(html_str)self.label.setPos(mousePoint.x(),mousePoint.y())self.vLine.setPos(mousePoint.x())self.hLine.setPos(mousePoint.y())self.vLine2.setPos(mousePoint.x())passdef mouseMoved2(self,evt):pos = evt[0]if self.pw2.sceneBoundingRect().contains(pos):mousePoint = self.vb2.mapSceneToView(pos)index = int(mousePoint.x())if index >= 0 and index < len(self.current_whole_data2):target_data = self.current_whole_data2[index]html_str = ''for i, item in enumerate(self.whole_header2):html_str += f"<br/>{item}:{target_data[i]}"self.label2.setHtml(html_str)self.label2.setPos(mousePoint.x(), mousePoint.y())self.vLine2.setPos(mousePoint.x())self.hLine2.setPos(mousePoint.y())self.vLine.setPos(mousePoint.x())passdef mouseClicked(self,evt):passdef updateViews(self):pass# 图形操作之外def duration_sel_btn_clicked(self):'''边界选择'''left_point = self.left_point.date().toString('yyyy-MM-dd')right_point = self.right_point.date().toString('yyyy-MM-dd')df = self.whole_df.copy()df['o_date'] = pd.to_datetime(df['tradeDate'])self.current_whole_df = df.loc[(df['o_date']>=left_point) & (df['o_date']<=right_point)].copy()self.caculate_and_show_data()passdef duration_reset_btn_clicked(self):'''边界重置'''self.current_whole_df = self.whole_df.copy()self.caculate_and_show_data()passdef duration_combox_currentIndexChanged(self,cur_i:int):cur_txt = self.duration_combox.currentText()if cur_txt == self.please_select_str:returncur_code = self.duration_map[cur_txt]right_point = res_date_normal_str()if cur_code == 'a':left_point = res_current_year_first_day()elif cur_code == 'b':left_point = res_pre_year_first_day()elif cur_code == 'c':left_point = res_pre_two_year_first_day()else:returndf = self.whole_df.copy()df['o_date'] = pd.to_datetime(df['tradeDate'])self.current_whole_df = df.loc[(df['o_date'] >= left_point) & (df['o_date'] <= right_point)].copy()self.caculate_and_show_data()passdef func_combox_currentIndexChanged(self,cur_i:int):cur_txt = self.func_combox.currentText()if cur_txt == self.please_select_str:returncur_code = self.func_map[cur_txt]df = self.current_whole_df.copy()if cur_code == 'a':# 多头排列 startdf['count'] = range(len(df))df['duo'] = 0df.loc[(df['ma5'] > df['ma10']) & (df['ma10'] > df['ma20']) & (df['ma20'] > df['ma30']) & (df['ma30'] > df['ma60']), 'duo'] = 1df['ext_0'] = df['duo'] - df['duo'].shift(1)df['ext_1'] = df['duo'] - df['duo'].shift(-1)duo_pre_df = df.loc[df['ext_0'] == 1].copy()duo_fix_df = df.loc[df['ext_1'] == 1].copy()pre_count_list = duo_pre_df['count'].values.tolist()pre_date_list = duo_pre_df['tradeDate'].values.tolist()fix_count_list = duo_fix_df['count'].values.tolist()fix_date_list = duo_fix_df['tradeDate'].values.tolist()if fix_count_list[0]<pre_count_list[0]:fix_count_list = fix_count_list[1:]fix_date_list = fix_date_list[1:]if fix_count_list[-1]<pre_count_list[-1]:pre_count_list = pre_count_list[0:-1]pre_date_list = pre_date_list[0:-1]font = QtGui.QFont()font.setPixelSize(9)for i, item in enumerate(fix_count_list):# 排除掉多头排列三日以下的区间if item-pre_count_list[i] <= 3:continuelr = pg.LinearRegionItem([pre_count_list[i], item],movable=False,brush=(self.color_up[0],self.color_up[1],self.color_up[2],50))lr.setZValue(-100)label_l = pg.InfLineLabel(lr.lines[0],pre_date_list[i],position=0.90,rotateAxis=(1,0),anchor=(1,1),color=self.color_up)label_l.setFont(font)label_r = pg.InfLineLabel(lr.lines[1],fix_date_list[i],position=0.90,rotateAxis=(1,0),anchor=(1,1),color=self.color_up)label_r.setFont(font)self.func_item_list.append(lr)self.pw.addItem(lr)# 多头排列 endpasselif cur_code == 'b':# 空头排列 startdf['count'] = range(len(df))df['duo'] = 0df.loc[(df['ma5'] < df['ma10']) & (df['ma10'] < df['ma20']) & (df['ma20'] < df['ma30']) & (df['ma30'] < df['ma60']), 'duo'] = 1df['ext_0'] = df['duo'] - df['duo'].shift(1)df['ext_1'] = df['duo'] - df['duo'].shift(-1)duo_pre_df = df.loc[df['ext_0'] == 1].copy()duo_fix_df = df.loc[df['ext_1'] == 1].copy()pre_count_list = duo_pre_df['count'].values.tolist()pre_date_list = duo_pre_df['tradeDate'].values.tolist()fix_count_list = duo_fix_df['count'].values.tolist()fix_date_list = duo_fix_df['tradeDate'].values.tolist()if fix_count_list[0] < pre_count_list[0]:fix_count_list = fix_count_list[1:]fix_date_list = fix_date_list[1:]if fix_count_list[-1] < pre_count_list[-1]:pre_count_list = pre_count_list[0:-1]pre_date_list = pre_date_list[0:-1]font = QtGui.QFont()font.setPixelSize(9)for i, item in enumerate(fix_count_list):# 排除掉空头排列三日以下的区间if item - pre_count_list[i] <= 3:continuelr = pg.LinearRegionItem([pre_count_list[i], item], movable=False, brush=(self.color_down[0],self.color_down[1],self.color_down[2], 50))lr.setZValue(-100)label_l = pg.InfLineLabel(lr.lines[0], pre_date_list[i], position=0.90, rotateAxis=(1, 0),anchor=(1, 1), color=self.color_down)label_l.setFont(font)label_r = pg.InfLineLabel(lr.lines[1], fix_date_list[i], position=0.90, rotateAxis=(1, 0),anchor=(1, 1), color=self.color_down)label_r.setFont(font)self.func_item_list.append(lr)self.pw.addItem(lr)# 空头排列 endpasselif cur_code == 'c':# 20日线斜率为正 startdf['count'] = range(len(df))df['ext_0'] = df['ma20'] - df['ma20'].shift(1)df['ext_1'] = 0df.loc[df['ext_0']>0,'ext_1'] = 1df['ext_2'] = df['ext_1'] - df['ext_1'].shift(1)df['ext_3'] = df['ext_1'] - df['ext_1'].shift(-1)duo_pre_df = df.loc[df['ext_2'] == 1].copy()duo_fix_df = df.loc[df['ext_3'] == 1].copy()pre_count_list = duo_pre_df['count'].values.tolist()pre_date_list = duo_pre_df['tradeDate'].values.tolist()fix_count_list = duo_fix_df['count'].values.tolist()fix_date_list = duo_fix_df['tradeDate'].values.tolist()if fix_count_list[0] < pre_count_list[0]:fix_count_list = fix_count_list[1:]fix_date_list = fix_date_list[1:]if fix_count_list[-1] < pre_count_list[-1]:pre_count_list = pre_count_list[0:-1]pre_date_list = pre_date_list[0:-1]font = QtGui.QFont()font.setPixelSize(9)for i, item in enumerate(fix_count_list):# if item - pre_count_list[i] <= 3:#     continuelr = pg.LinearRegionItem([pre_count_list[i], item], movable=False, brush=(self.color_up[0], self.color_up[1], self.color_up[2], 50))lr.setZValue(-100)label_l = pg.InfLineLabel(lr.lines[0], pre_date_list[i], position=0.90, rotateAxis=(1, 0),anchor=(1, 1), color=self.color_up)label_l.setFont(font)label_r = pg.InfLineLabel(lr.lines[1], fix_date_list[i], position=0.90, rotateAxis=(1, 0),anchor=(1, 1), color=self.color_up)label_r.setFont(font)self.func_item_list.append(lr)self.pw.addItem(lr)# 20日线斜率为正 endpasselif cur_code == 'd':# 阶段最高点最低点 start# 最高点df['ext_0'] = df['closePrice'].cummax()df['ext_1'] = df['ext_0'] - df['ext_0'].shift(1)# 最低点df['ext_2'] = df['closePrice'].cummin()df['ext_3'] = df['ext_2'] - df['ext_2'].shift(1)max_point_df = df.loc[df['ext_1']==0].copy()min_point_df = df.loc[df['ext_3']==0].copy()# 阶段最高点最低点 endpasspassdef clear_func_btn_clicked(self):for item in self.func_item_list:self.pw.removeItem(item)self.func_item_list.clear()passpass

使用

if __name__ == '__main__':# 先显示蜡烛图# 600660  福耀玻璃import pandas as pdimport mathimport talibdf = pd.read_csv('E:/temp005/600660.csv',encoding='utf-8')# 删除停牌的数据df = df.loc[df['openPrice']>0].copy()df['openPrice'] = df['openPrice']*df['accumAdjFactor']df['closePrice'] = df['closePrice']*df['accumAdjFactor']df['highestPrice'] = df['highestPrice']*df['accumAdjFactor']df['lowestPrice'] = df['lowestPrice']*df['accumAdjFactor']# 计算指标close_list = df['closePrice']df['rsi'] = talib.RSI(close_list,timeperiod=14)whole_pd_header = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice']whole_pd_header2 = ['tradeDate','rsi']line_data = {'title_str':'福耀玻璃','whole_header':['日期','收盘价','开盘价','最高价','最低价'],'whole_header2':['日期','rsi'],'whole_pd_header':whole_pd_header,'whole_pd_header2':whole_pd_header2,'whole_df':df}app = QtWidgets.QApplication(sys.argv)t_win = PyQtGraphLineWidget()t_win.show()t_win.set_data(line_data)sys.exit(app.exec_())pass

鼠标滚动能同步缩放

数据

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

PyQt5_pyqtgraph股票RSI指标相关推荐

以股票RSI指标为例，学习Python发送邮件功能（含RSI指标确定卖点策略）
本人之前写过若干"给程序员加财商"的系列文,目的是通过股票案例讲述Python知识点,让大家在学习Python的同时还能掌握相关的股票知识,所谓一举两得. 在之前的系列文里,大家能 ...
PyQt5_pyqtgraph股票SAR指标
股票K线图与SAR指标横轴同步缩放显示,SAR显示美式K线图和红绿豆,方便同步查看股票价格与SAR指标之间的关系,在使用代码辅助分析时,通过该控件可以方便所得分析结果可见,协助分析. 目录效果代码 ...
ddz指标最佳数值_一个良心操盘手的肺腑之言：炒股10年从未被套，只依靠RSI指标！...
相对强弱指数RSI是目前流行最广.使用最多的技术指标之一,它是由技术分析大师威尔德创造的. RSI可应用于股票.期货和外汇等市场.股票指数是由股市中的供求关系决定的,供应和需求达到平衡,指数才能稳定. ...
RSI指标使用技巧和参数设置
相对强弱指标RSI,是用来判断买卖双方力量强弱的技术指标,由韦尔斯-王尔德提出.外汇交易中有很多常见的指标,rsi指标就是其中一个.那么rsi指标参数怎么设置?如果您不清楚外汇rsi指标参数设置的方法 ...
RSI指标的秘密用法
相对强弱指数RSI是根据一定时期内上涨点数和涨跌点数之和的比率制作出的一种技术曲线.能够反映出市场在一定时期内的景气程度.由威尔斯.威尔德(Welles Wilder)最早应用于期货买卖,后来人们发现 ...
RSI指标各种买卖信号公式源码分享
相对强弱指数RSI是目前流行最广,使用最多的技术指标之一,他是技术分析大师威尔德创造的.RSI之所以流行最广,是因为他不但可以应用在股票市场,在期货和外汇市场也有较好的应用. RSI是以一特定时期内股 ...
RSI指标的原理和计算方法
相对强弱指标RSI又叫力度指标,其英文全称为"Relative Strength Index",由威尔斯•魏尔德﹝Welles Wilder﹞所创造的,是目前股市技术分析中比较常用 ...
量化投资常用技能——指标篇3：详解RSI指标，及其代码实现和绘图
量化投资常用技能系列文章目录我们已经介绍了三篇关于量化投资方面绘图的文章和两篇指标类的推导和介绍的文章,大家有兴趣可以了解一下绘图篇量化投资常用技能--绘图篇 1:绘制股票收盘价格曲线和och ...
常用技术指标之一文读懂RSI指标
什么是RSI指标? RSI (Relative Strength Index) 中文名又叫相对强弱指标,由韦尔斯.怀尔德(Welles Wilder)首创,发表在他的<技术交易系统新思路> ...

PyQt5_pyqtgraph股票RSI指标

效果

代码

使用

数据

PyQt5_pyqtgraph股票RSI指标相关推荐

最新文章

热门文章