Search code examples
pythonpyqt5pyqtgraph

PyQtGraph - How to set intervals of axis


Below is a function I have written that create a graph based on a list of tuples of scores (scores and their frequency)

def initialise_chart(self, scores):

    pg.setConfigOption("background", "w")
    pg.setConfigOption("foreground", "k")

    results_graph = pg.PlotWidget()
    self.chart_results.addWidget(results_graph)
    results_graph.plot([i[0] for i in scores], [i[1] for i in scores], pen={'color': "#006eb4"})
    results_graph.setLabels(left="Frequency", bottom="Scores")
    results_graph.setXRange(0, self.max_mark, padding=0)

This produces the following graph: enter image description here

Is there any way to set the intervals of the y-axis so that the numbers go up in steps of 1, but the range is still autoscaled? eg. the only number displayed on the example graph's y-axis would be 0, 1, 2


Solution

  • You must change the ticks on the AxisItem, by example:

    import pyqtgraph as pg
    import numpy as np
    from pyqtgraph.Qt import QtCore, QtGui
    
    app = pg.mkQApp()
    
    pw = pg.PlotWidget(title="Example")
    x = np.arange(20)
    y = x**2/150
    pw.plot(x=x, y=y, symbol='o')
    pw.show()
    pw.setWindowTitle('Example')
    
    ax = pw.getAxis('bottom')  # This is the trick
    dx = [(value, str(value)) for value in list((range(int(min(x.tolist())), int(max(x.tolist())+1))))]
    ax.setTicks([dx, []])
    
    ay = pw.getAxis('left')  # This is the trick
    dy = [(value, str(value)) for value in list((range(int(min(y.tolist())), int(max(y.tolist())+1))))]
    ay.setTicks([dy, []])
    
    if __name__ == '__main__':
        import sys
    
        if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
            QtGui.QApplication.instance().exec_()
    

    Before:

    enter image description here

    After:

    enter image description here