python获取股票每天k线(使用PyQtGraph绘制精美的股票行情K线图的示例代码)
类别:脚本大全 浏览量:316
时间:2022-01-18 01:48:47 python获取股票每天k线
使用PyQtGraph绘制精美的股票行情K线图的示例代码pyqtgraph是python平台上一种功能强大的2d/3d绘图库,相对于matplotlib库,由于其在内部实现方式上,使用了高速计算的numpy信号处理库以及qt的graphicsview框架,因此它在大数据量的处理及快速显示方面有着天然的优势,非常适合于需要快速绘图更新、视频或实时交互性的操作场合,在数学、科学和工程领域都有着广泛的应用。
k线图介绍
对于股票交易者来讲,k线图是弄清股票一段时间走势的一种最基本的图形工具,k线分为阳线和阴线,阳线和阴线都包含了开盘价、收盘价、最高价和最低价,一般k线如下图所示:
当收盘价大于开盘价时,称为阳线,在图形上一般用红色表示,反之,当收盘价低于开盘价时,称为阴线,在图形上一般用绿色表示。由于其形状颇似一根根蜡烛,k线图有时也叫做蜡烛图。
|
# -*- coding: utf-8 -*- # form implementation generated from reading ui file 'qwidget_plot.ui' # # created by: pyqt4 ui code generator 4.11.4 # # warning! all changes made in this file will be lost! import sys reload (sys) sys.setdefaultencoding( 'utf-8' ) from pyqt4 import qtcore, qtgui import datetime import pyqtgraph as pg import tushare as ts try : _fromutf8 = qtcore.qstring.fromutf8 except attributeerror: def _fromutf8(s): return s try : _encoding = qtgui.qapplication.unicodeutf8 def _translate(context, text, disambig): return qtgui.qapplication.translate(context, text, disambig, _encoding) except attributeerror: def _translate(context, text, disambig): return qtgui.qapplication.translate(context, text, disambig) class ui_mainwindow( object ): def setupui( self , mainwindow): mainwindow.setobjectname(_fromutf8( "mainwindow" )) mainwindow.resize( 800 , 600 ) self .centralwidget = qtgui.qwidget(mainwindow) self .centralwidget.setobjectname(_fromutf8( "centralwidget" )) self .verticallayout_2 = qtgui.qvboxlayout( self .centralwidget) self .verticallayout_2.setobjectname(_fromutf8( "verticallayout" )) self .verticallayout_2.setobjectname(_fromutf8( "verticallayout_2" )) self .verticallayout_2.setcontentsmargins( 0 , 0 , 0 , 0 ) mainwindow.setcentralwidget( self .centralwidget) self .menubar = qtgui.qmenubar(mainwindow) self .menubar.setgeometry(qtcore.qrect( 0 , 0 , 800 , 31 )) self .menubar.setobjectname(_fromutf8( "menubar" )) mainwindow.setmenubar( self .menubar) self .drawchart = drawchart(ktype = 'd' ) self .verticallayout_2.addwidget( self .drawchart.pyqtgraphdrawchart()) self .retranslateui(mainwindow) qtcore.qmetaobject.connectslotsbyname(mainwindow) def retranslateui( self , mainwindow): mainwindow.setwindowtitle(_translate( "mainwindow" , "mainwindow" , none)) class drawchart(): def __init__( self , code = 'sz50' , start = str (datetime.date.today() - datetime.timedelta(days = 200 )), end = str (datetime.date.today() + datetime.timedelta(days = 1 )), ktype = 'd' ): self .code = code self .start = start self .end = end self .ktype = ktype self .data_list, self .t = self .getdata() def pyqtgraphdrawchart( self ): try : self .item = candlestickitem( self .data_list) self .xdict = { 0 : str ( self .hist_data.index[ 0 ]).replace( '-' , '/' ), int (( self .t + 1 ) / 2 ) - 1 : str ( self .hist_data.index[ int (( self .t + 1 ) / 2 )]).replace( '-' , '/' ), self .t - 1 : str ( self .hist_data.index[ - 1 ]).replace( '-' , '/' )} self .stringaxis = pg.axisitem(orientation = 'bottom' ) self .stringaxis.setticks([ self .xdict.items()]) self .plt = pg.plotwidget(axisitems = { 'bottom' : self .stringaxis}, enablemenu = false) self .plt.additem( self .item) # self.plt.showgrid(x=true, y=true) return self .plt except : return pg.plotwidget() def getdata( self ): self .start = str (datetime.date.today() - datetime.timedelta(days = 150 )) self .end = str (datetime.date.today() + datetime.timedelta(days = 1 )) self .hist_data = ts.get_hist_data( self .code, self .start, self .end, self .ktype).sort_index()[ - 300 : - 1 ] data_list = [] t = 0 for dates, row in self .hist_data.iterrows(): open , high, close, low, volume, price_change, p_change, ma5, ma10, ma20 = row[: 10 ] datas = (t, open , close, low, high, volume, price_change, p_change, ma5, ma10, ma20) data_list.append(datas) t + = 1 return data_list, t class candlestickitem(pg.graphicsobject): def __init__( self , data): pg.graphicsobject.__init__( self ) self .data = data self .generatepicture() def generatepicture( self ): self .picture = qtgui.qpicture() p = qtgui.qpainter( self .picture) p.setpen(pg.mkpen( 'w' )) w = ( self .data[ 1 ][ 0 ] - self .data[ 0 ][ 0 ]) / 3. prema5 = 0 prema10 = 0 prema20 = 0 for (t, open , close, min , max , volume, price_change, p_change, ma5, ma10, ma20) in self .data: if open > close: p.setpen(pg.mkpen( 'g' )) p.setbrush(pg.mkbrush( 'g' )) else : p.setpen(pg.mkpen( 'r' )) p.setbrush(pg.mkbrush( 'r' )) p.drawline(qtcore.qpointf(t, min ), qtcore.qpointf(t, max )) p.drawrect(qtcore.qrectf(t - w, open , w * 2 , close - open )) if prema5 ! = 0 : p.setpen(pg.mkpen( 'w' )) p.setbrush(pg.mkbrush( 'w' )) p.drawline(qtcore.qpointf(t - 1 , prema5), qtcore.qpointf(t, ma5)) prema5 = ma5 if prema10 ! = 0 : p.setpen(pg.mkpen( 'c' )) p.setbrush(pg.mkbrush( 'c' )) p.drawline(qtcore.qpointf(t - 1 , prema10), qtcore.qpointf(t, ma10)) prema10 = ma10 if prema20 ! = 0 : p.setpen(pg.mkpen( 'm' )) p.setbrush(pg.mkbrush( 'm' )) p.drawline(qtcore.qpointf(t - 1 , prema20), qtcore.qpointf(t, ma20)) prema20 = ma20 p.end() def paint( self , p, * args): p.drawpicture( 0 , 0 , self .picture) def boundingrect( self ): return qtcore.qrectf( self .picture.boundingrect()) if __name__ = = "__main__" : import sys app = qtgui.qapplication(sys.argv) mainwindow = qtgui.qmainwindow() ui = ui_mainwindow() ui.setupui(mainwindow) mainwindow.show() sys.exit(app.exec_()) |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持开心学习网。
原文链接:https://blog.csdn.net/otter1010/article/details/83719709
热门推荐
- dedecms本地搭建(dedecms blog博客模板安装说明)
- 如何布局网站的广告位
- mysqldump导入导出(MySQL官方导出工具mysqlpump的使用)
- axios调用接口(如何实现axios的自定义适配器adapter)
- js如何生成随机数
- python企业微信录信息(python获取微信企业号打卡数据并生成windows计划任务)
- spring-boot 内置tomcat启动(centos环境下使用tomcat 部署SpringBoot的war包)
- css五种使用方法(CSS 的加载及加载顺序简介)
- mysql获取当前日期时间(MySQL DATEDIFF函数获取两个日期的时间间隔的方法)
- 新加坡服务器目前负载(新加坡服务器好不好?新加坡服务器有什么优势?)