PyQt5 그래프 그리기 - PyQtGraph 예제

PyQtGraph 라이브러리를 사용하여 PyQt에서 그래프 그리는 예제 코드입니다.

 

 

2024. 2. 5 최초작성

다음 2개의 패키지를 설치해야 합니다.

 

pip install pyqtgraph PyQt5  

디폴트 배경색은 검은색이기 때문에 흰색으로 변경할 필요가 있습니다.

아래 예제 코드에서는 setBackground 메소드에 w를 지정하여 배경색이 흰색이 됩니다. 

 

import pyqtgraph as pg from PyQt5 import QtWidgets class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")        time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.plot(time, temperature) app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

 

 

 

배경색으로 많이 지정하는 배경색은 다음과 같습니다. Matplotlib에서 사용되는 방식으로 색을 지정합니다. 

 

16진수 형식으로 색을 지정할 수도 있습니다.

 

self.plot_graph.setBackground("#fff8dc")

RGB 또는 RGBA 형식으로 지정할 수 도 있습니다.  RGBA는 RGB에 색의 투명도인 알파 채널을 추가한 것입니다.

 

RGB를 사용하여 색을 지정한 예입니다.

 

self.plot_graph.setBackground((255, 160, 122))

RGBA를 사용하여 색지정한 예입니다. 앞에서 지정한 RGB 값에 알파채널값을 30으로 해봅니다. 배경색이 투명해진걸 볼 수 있습니다.

 

self.plot_graph.setBackground((255, 160, 122, 30))

16진수(HEX) 색 / RGB 색 지정 예는 아래 표를 확인하세요. 

( 이미지 출처 - https://mayburilova.wordpress.com/2010/11/02/hexadecimal-colors/ )

 

그래프 선 색깔을 빨간색으로 지정하는 예제코드입니다. 선 색을 ( r, g, b)로 지정합니다. 

 

import pyqtgraph as pg from PyQt5 import QtWidgets class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0))         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.plot(time, temperature, pen=pen) app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

선 굵기를 5로 조정하고 선의 타입을 대시로 변경합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.plot(time, temperature, pen=pen) app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

 

사용가능한 선 타입은 다음과 같습니다. ( 이미지 출처 https://www.pythonguis.com/tutorials/plotting-pyqtgraph/  )

 

데이터 포인트마다 마커를 추가합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.plot(time, temperature, pen=pen, symbol="+") app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

마커를 커스터마이즈한 예제입니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.plot(time, temperature, pen=pen, symbol="+", symbolSize=20, symbolBrush="b") app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

+마커 외에 사용가능한 마커 종류입니다. ( 이미지 출처 https://www.pythonguis.com/tutorials/plotting-pyqtgraph/  )

 

 

그래프에 제목을 추가하고 좌표축에 라벨을 추가합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt") # 그래프 제목         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles) # y축 라벨         self.plot_graph.setLabel("bottom", "Time (min)", **styles) # x축 라벨         self.plot_graph.plot(time, temperature, pen=pen, symbol="+", symbolSize=20, symbolBrush="b") app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

범례가 보이려면 그래프를 그려주는 plot 메소드에  name 속성을 추가해야 합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt")         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles)         self.plot_graph.setLabel("bottom", "Time (min)", **styles)         self.plot_graph.addLegend()         self.plot_graph.plot(time, temperature, pen=pen, name="Temperature Sensor", symbol="+", symbolSize=20, symbolBrush="b") # name을 추가해야 합니다. app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

그래프가 그려지는 배경에 x축과 y축 그리드를 추가합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt")         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles)         self.plot_graph.setLabel("bottom", "Time (min)", **styles)         self.plot_graph.addLegend()         self.plot_graph.showGrid(x=True, y=True)         self.plot_graph.plot(time, temperature, pen=pen, name="Temperature Sensor", symbol="+", symbolSize=20, symbolBrush="b") # name을 추가해야 합니다. app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

축 범위를 지정합니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt")         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles)         self.plot_graph.setLabel("bottom", "Time (min)", **styles)         self.plot_graph.addLegend()         self.plot_graph.showGrid(x=True, y=True)         self.plot_graph.setXRange(1, 10)         self.plot_graph.setYRange(20, 40)         self.plot_graph.plot(time, temperature, pen=pen, name="Temperature Sensor", symbol="+", symbolSize=20, symbolBrush="b") # name을 추가해야 합니다. app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

그래프 여러개를 그려주는 예제 코드입니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         time = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt")         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles)         self.plot_graph.setLabel("bottom", "Time (min)", **styles)         self.plot_graph.addLegend()         self.plot_graph.showGrid(x=True, y=True)         self.plot_graph.setXRange(1, 10)         self.plot_graph.setYRange(20, 40)         temperature1 = [30, 32, 34, 32, 33, 31, 29, 32, 35, 30]         temperature2 = [33, 34, 35, 36, 31, 33, 37, 29, 23, 21]         pen1 = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         pen2 = pg.mkPen(color=(0, 0, 255), width=5, style=QtCore.Qt.DashLine)         self.plot_graph.plot(time, temperature1, pen=pen1, name="Temperature Sensor 1", symbol="+", symbolSize=20, symbolBrush="r") # name을 추가해야 합니다.         self.plot_graph.plot(time, temperature2, pen=pen2, name="Temperature Sensor 2", symbol="+", symbolSize=20, symbolBrush="b") # name을 추가해야 합니다. app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

동적으로 그래프를 그려주는 예제코드입니다.

 

import pyqtgraph as pg from PyQt5 import QtWidgets, QtCore from random import randint from pyqtgraph import LegendItem class MainWindow(QtWidgets.QMainWindow):     def __init__(self):         super().__init__()         # Temperature vs time plot         self.plot_graph = pg.PlotWidget()         self.setCentralWidget(self.plot_graph)         self.plot_graph.setBackground("w")         self.pen = pg.mkPen(color=(255, 0, 0), width=5, style=QtCore.Qt.DashLine)         # 좀 더 많은 데이터를 생성합니다.         self.time = list(range(50))         self.temperature = [randint(20, 40) for _ in range(50)]         self.plot_graph.setTitle("Temperature vs Time", color="b", size="20pt")         styles = {"color": "red", "font-size": "18px"}         self.plot_graph.setLabel("left", "Temperature (°C)", **styles)         self.plot_graph.setLabel("bottom", "Time (min)", **styles)         # self.plot_graph.addLegend()         self.plot_graph.showGrid(x=True, y=True)         # self.plot_graph.setXRange(1, 10)         # self.plot_graph.setYRange(20, 40)         self.graph = self.plot_graph.plot([], [], pen=self.pen, name="Temperature Sensor", symbol="+", symbolSize=20, symbolBrush="b") # name을 추가해야 합니다.         # 300ms마다 타이머에서 지정한 함수가 호출됩니다.         self.timer = QtCore.QTimer()         self.timer.setInterval(300)         self.timer.timeout.connect(self.update_plot)         self.timer.start()         self.i = 1     def update_plot(self):         time = self.time[:self.i]         temperature = self.temperature[:self.i]         self.graph.setData(time, temperature)                 self.plot_graph.setXRange(self.i-10, self.i)                 self.i = self.i + 1 app = QtWidgets.QApplication([]) main = MainWindow() main.show() app.exec()

참고

 

https://www.pythonguis.com/tutorials/plotting-pyqtgraph/