您的位置:首页 > 脚本大全 > > 正文

python获取股票每天k线(使用PyQtGraph绘制精美的股票行情K线图的示例代码)

更多 时间:2022-01-18 01:48:47 类别:脚本大全 浏览量:316

python获取股票每天k线

使用PyQtGraph绘制精美的股票行情K线图的示例代码

pyqtgraph是python平台上一种功能强大的2d/3d绘图库,相对于matplotlib库,由于其在内部实现方式上,使用了高速计算的numpy信号处理库以及qt的graphicsview框架,因此它在大数据量的处理及快速显示方面有着天然的优势,非常适合于需要快速绘图更新、视频或实时交互性的操作场合,在数学、科学和工程领域都有着广泛的应用。

k线图介绍

对于股票交易者来讲,k线图是弄清股票一段时间走势的一种最基本的图形工具,k线分为阳线和阴线,阳线和阴线都包含了开盘价、收盘价、最高价和最低价,一般k线如下图所示:

当收盘价大于开盘价时,称为阳线,在图形上一般用红色表示,反之,当收盘价低于开盘价时,称为阴线,在图形上一般用绿色表示。由于其形状颇似一根根蜡烛,k线图有时也叫做蜡烛图。

  • ?
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102
  • 103
  • 104
  • 105
  • 106
  • 107
  • 108
  • 109
  • 110
  • 111
  • 112
  • 113
  • 114
  • 115
  • 116
  • 117
  • 118
  • 119
  • 120
  • 121
  • 122
  • 123
  • 124
  • 125
  • 126
  • 127
  • 128
  • 129
  • 130
  • 131
  • 132
  • 133
  • 134
  • 135
  • 136
  • 137
  • 138
  • 139
  • 140
  • 141
  • 142
  • 143
  • 144
  • 145
  • # -*- 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_())
  • python获取股票每天k线(使用PyQtGraph绘制精美的股票行情K线图的示例代码)

    以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持开心学习网。

    原文链接:https://blog.csdn.net/otter1010/article/details/83719709

    标签:K线图 PyQtGraph