python pyqt绘制直方图

Posted 2020-10-16 Pavilion

# -*- coding: utf-8 -*-
"""
In this example we draw two different kinds of histogram.
"""

from qtpy import QtWidgets, QtGui, QtCore
from qtpy.QtWidgets import QApplication, QWidget
import datetime as dt
from vnpy.trader import *
from vnpy.trader.uiKLine  import *
from vnpy.trader.widget.crosshairTool import  CrosshairTool
import pyqtgraph as pg
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.mlab as mlab
import sys
class HistogramLUTWidget(QWidget):
    def __init__(self, data, parent=None):
        self.parent = parent
        self.data = data
        super(HistogramLUTWidget, self).__init__(parent)  # 继承
    # 界面布局
        self.pw=pg.PlotWidget()

        self.lay_KL = pg.GraphicsLayout(border=(100, 100, 100))
        self.lay_KL.setContentsMargins(10, 10, 10, 10)
        self.lay_KL.setSpacing(0)
        self.lay_KL.setBorder(color=(255, 255, 255, 255), width=0.8)
        self.lay_KL.setZValue(0)
        self.pw.setCentralItem(self.lay_KL)
    #     # 设置横坐标
    #     xdict = dict(enumerate(self.data["timeList"]))
    #     self.axisTime = MyStringAxis(xdict, orientation=‘bottom‘)
    #     # 初始化子图
        self.initplotpnl()
        # # 注册十字光标
        x=[]
        viwe = self.plt
        self.crosshair = CrosshairTool(self.pw, x, viwe, self)
        # 设置界面
        self.vb = QtWidgets.QVBoxLayout()
        self.vb.addWidget(self.pw)
        self.setLayout(self.vb)
    # ----------------------------------------------------------------------

    def initplotpnl(self, xmin=0, xmax=-1):
        vals = np.hstack(self.data)
        xMin = min(self.data)
        xMax = max(self.data)
        #histogram
        #y, x = np.histogram(vals, bins=np.linspace(xMin,xMax, 100),normed=False,density=True)
        # #hist
        nx, xbins, ptchs = plt.hist(self.data, bins=50, normed=True, facecolor=‘black‘, edgecolor=‘black‘,alpha=1,histtype = ‘bar‘)
        histo = plt.hist(self.data, 50)
        vb = CustomViewBox()
        # self.plt=pg.PlotItem(viewBox=vb,  axisItems={‘bottom‘: self.axisTime})#设置x轴
        self.plt = pg.PlotItem(viewBox=vb)
        leng_ = len(self.data)
        self.plt.setRange(xRange=(0, leng_), yRange=(xMin, xMax))
        # #histogram
        # self.plt.plot(x, y, stepMode=True, fillLevel=0, brush=(0, 0, 255, 50),size=0.4)#绘制统计所得到的概率密度,直方图，没有边界分割线
        ##hist  bar
        width = xbins[1] - xbins[0]  # Width of each bin.
        #self.plt.plot(x,y,width=width)# 绘制统计所得到的概率密度,线形图
        self.pi=pg.BarGraphItem(x=histo[1][0:50],height=histo[0],width=width, color=‘y‘,)# 绘制统计所得到的概率密度,bar图
        self.plt.addItem(self.pi)
        self.lay_KL.addItem(self.plt)

    def refresh(self):
        """
        刷新子图的现实范围
        """
        leng_ = len(self.data)
        xMin = min(self.data)
        xMax = max(self.data)
        self.plt.setRange(xRange=(0, leng_), yRange=(xMin, xMax))


if __name__ == ‘__main__‘:
    app = QApplication(sys.argv)
    ex = HistogramLUTWidget()
    sys.exit(app.exec_())

####才接触，项目急没有细研究，有错误的改正的地方，望大家不吝指出