Draw a graph with PyQtGraph Part 6-Displaying a legend

Part 6 Displaying the legend

I don't really like the PyQtGraph legend because it's inflexible. ~~ You should not have been able to change the background color etc. ~~ ↑ You pointed out in the comments. It is possible to change the background color itself by modifying the library (pyqtgraph \ graphicsItems \ LegendItem.py). See the comments section for details.

6.png

import sys
from PySide.QtCore import *
from PySide.QtGui import *
import pyqtgraph as pg


class GraphWindow(QMainWindow):

    def __init__(self, parent = None):
        
        super(GraphWindow, self).__init__(parent)
        
        pw = pg.PlotWidget(viewBox = pg.ViewBox(border = pg.mkPen(color='#000000'),
                                                invertX = False, invertY = False))

        self.setCentralWidget(pw)

        pw.setBackground("#FFFFFF00")

        pw.setMinimumSize(500, 400)
        pw.setMaximumSize(500, 400)

        p1 = pw.plotItem
        p2 = pg.ViewBox()
        p3 = pg.ViewBox()
        ax3 = pg.AxisItem(orientation = 'right')
        self.setGraphFrameFont(p1, p2, p3, ax3)
        self.setGraphMultipleAxis(p1, p2, p3, ax3)

        p1.setLabels(bottom = "X Axis", 
                     left = "Y1 Axis", 
                     right = "Y2 Axis")
        ax3.setLabel("Y3 Axis")

        fontCss = {'font-family': "Times New Roman,Meiryo", 'font-size': '10.5pt', "color": "black"}
        p1.getAxis('bottom').setLabel(**fontCss)
        fontCss["color"] = "red"
        p1.getAxis('left').setLabel(**fontCss)
        fontCss["color"] = "blue"
        p1.getAxis('right').setLabel(**fontCss)
        fontCss["color"] = "green"
        ax3.setLabel(**fontCss)

        p1.setRange(xRange = (-2, 6), yRange = (-2, 6), padding = 0)
        p2.setRange(yRange = (-1, 5), padding = 0)
        p3.setRange(yRange = (0, 4), padding = 0)

        p1.getAxis('bottom').setTickSpacing(major = 2.5, minor = 1)
        p1.getAxis('left').setTickSpacing(major = 2.5, minor = 1)
        p1.getAxis('right').setTickSpacing(major = 2.5, minor = 1)
        ax3.setTickSpacing(major = 2.5, minor = 1)
        
        p1.addItem(pg.PlotCurveItem(x = [0, 1, 2, 3 ,4], 
                                    y = [0, 1, 2, 3 ,4], 
                                    pen = pg.mkPen(color = "r", style = Qt.SolidLine), 
                                    name = "plot1", 
                                    antialias = True))
        p2.addItem(pg.ScatterPlotItem(x = [0, 1, 2, 3 ,4], 
                                      y = [4, 3, 2, 1, 0], 
                                      symbol = "x", 
                                      pen = pg.mkPen(None), 
                                      name = "plot2", 
                                      brush = pg.mkBrush("b"),
                                      size = 7.5,
                                      antialias = True))
        p3.addItem(pg.PlotCurveItem(x = [0, 1, 2, 3 ,4], 
                                    y = [0, 4, 0, 4, 0], 
                                    pen = pg.mkPen(color = "g", style = Qt.DashLine),
                                    name = "plot3", 
                                    antialias = True))

        ##6 Set and display the legend
        legend = p1.addLegend(offset = (10, 10))
        fontCssLegend = '<style type="text/css"> p {font-family: Times New Roman,Meiryo; font-size: 10.5pt; color: "#000000"} </style>'
        legend.addItem(pg.PlotCurveItem(pen = pg.mkPen(color = "r", style = Qt.SolidLine), antialias = True), 
                       name = fontCssLegend + '<p>plot1 (Plot 1)</p>')
        legend.addItem(pg.ScatterPlotItem(symbol = "x", pen = pg.mkPen(None), brush = pg.mkBrush("b"), size = 7.5, antialias = True), 
                       name = fontCssLegend + '<p>plot2 (Plot 2)</p>')
        legend.addItem(pg.PlotCurveItem(pen = pg.mkPen(color = "g", style = Qt.DashLine), antialias = True), 
                       name = fontCssLegend + '<p>plot3 (Plot 3)</p>')


    def setGraphFrameFont(self, p1, p2 = None, p3 = None, ax3 = None):
        font = QFont("Times New Roman", 7)

        p1.getAxis('bottom').setPen(pg.mkPen(color='#000000'))
        p1.getAxis('left').setPen(pg.mkPen(color='#000000'))
        p1.getAxis('bottom').setHeight(3.5 * 10.5)
        p1.getAxis('left').setWidth(4 * 10.5)

        #p1.getAxis('bottom').setLabel(**self.fontCss)
        p1.getAxis('bottom').tickFont = font
        #p1.getAxis('left').setLabel(**self.fontCss)
        p1.getAxis('left').tickFont = font

        if p2 != None:
            p1.getAxis('right').setPen(pg.mkPen(color='#000000'))
            p1.getAxis('right').setWidth(4 * 10.5)

            #p1.getAxis('right').setLabel(**self.fontCss)
            p1.getAxis('right').tickFont = font

        if p3 != None and ax3 != None:
            ax3.setPen(pg.mkPen(color='#000000'))
            ax3.setWidth(4 * 10.5)

            #ax3.setLabel(**self.fontCss)
            ax3.tickFont = font


    def setGraphMultipleAxis(self, p1, p2, p3 = None, ax3 = None):
        p1.showAxis('right')
        p1.scene().addItem(p2)
        p1.getAxis('right').linkToView(p2)
        p2.setXLink(p1)

        p2.sigRangeChanged.connect(lambda: p2.setGeometry(p1.vb.sceneBoundingRect()))
        
        if p3 != None and ax3 != None:
            spacer = QGraphicsWidget()
            spacer.setMaximumSize(15,15)
            p1.layout.addItem(spacer, 2, 3)

            p1.layout.addItem(ax3, 2, 4)
            p1.scene().addItem(p3)
            ax3.linkToView(p3)
            p3.setXLink(p1)
            
            p3.sigRangeChanged.connect(lambda: p3.setGeometry(p1.vb.sceneBoundingRect()))


if __name__ == '__main__':
    #Create a Qt Application
    app = QApplication(sys.argv)
    #Create and display form
    mainWin = GraphWindow()
    mainWin.show()
    #Start the main loop of Qt
    sys.exit(app.exec_())

Recommended Posts

Draw a graph with PyQtGraph Part 6-Displaying a legend
Draw a graph with PyQtGraph Part 1-Drawing
Draw a graph with PyQtGraph Part 3-PlotWidget settings
Draw a graph with PyQtGraph Part 4-PlotItem settings
Draw a graph with Julia + PyQtGraph (2)
Draw a graph with Julia + PyQtGraph (1)
Draw a graph with Julia + PyQtGraph (3)
Draw a graph with PyQtGraph Part 5-Increase the Y-axis
Draw a graph with PyQtGraph Part 2--Detailed plot settings
Draw a graph with networkx
Draw a loose graph with matplotlib
Draw a graph with pandas + XlsxWriter
Draw a graph with PySimple GUI
Draw a flat surface with a matplotlib 3d graph
Draw a graph with Japanese labels in Jupyter
Draw a graph by processing with Pandas groupby
[Python] Draw a directed graph with Dash Cytoscape
Draw a graph with matplotlib from a csv file
[Python] How to draw a line graph with Matplotlib
Create a heatmap with pyqtgraph
Study math with Python: Draw a sympy (scipy) graph with matplotlib
[Visualization] I want to draw a beautiful graph with Plotly
Play with a turtle with turtle graphics (Part 1)
Draw a beautiful circle with numpy
Let's make a graph with python! !!
Make a nice graph with plotly
Easily draw a map with matplotlib.basemap
[PyQt] Display a multi-axis graph with QtChart
How to draw a graph using Matplotlib
(Matplotlib) I want to draw a graph with a size specified in pixels
Draw a heart in Ruby with PyCall
Draw a heart in Python Part 2 (SymPy)
Draw a Mandelbrot set with Brainf * ck
How to draw a bar graph that summarizes multiple series with matplotlib
Draw a graph that can be moved around with HoloViews and Bokeh
Simply draw a graph by specifying a file
Create a graph with borders removed with matplotlib
Draw a graph of a quadratic function in Python
Tweet the weather forecast with a bot Part 2
Draw a "breast curved surface" in a 3D graph (1)
A memo that made a graph animated with plotly
[Python] Draw a Mickey Mouse with Turtle [Beginner]
Draw a "breast curved surface" in a 3D graph (2)
I made a random number graph with Numpy
Draw multiple photos in a graph from multiple folders
Draw a graph in Julia ... I tried a little analysis
I tried to draw a route map with Python
How to make a shooting game with toio (Part 1)
Visualize railway line data as a graph with Cytoscape 2
"My Graph Generation Application" by Python (PySide + PyQtGraph) Part 2
Forcibly draw something like a flowchart with Python, matplotlib
I want to manually create a legend with matplotlib
Create a 3D model viewer with PyQt5 and PyQtGraph
"My Graph Generation Application" by Python (PySide + PyQtGraph) Part 1
[Python] How to draw a scatter plot with Matplotlib
Data visualization with Python-It's too convenient to draw a graph by attribute with "Facet" at once