[python] PyQt에서의 데이터 시각화

30 Nov 2023

python

PyQt는 파이썬 프로그래밍 언어를 위한 툴킷으로, 다양한 GUI 응용 프로그램을 개발할 수 있도록 도와줍니다. PyQt를 사용하면 데이터 시각화도 쉽게 할 수 있습니다. 이 글에서는 PyQt를 사용하여 데이터를 시각화하는 방법에 대해 알아보겠습니다.

Matplotlib를 이용한 데이터 시각화

Matplotlib는 파이썬에서 가장 유명한 데이터 시각화 라이브러리 중 하나입니다. PyQt에서 Matplotlib를 사용하면 다양한 차트와 그래프를 생성할 수 있습니다. 아래는 간단한 예제 코드입니다.

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # Matplotlib Figure 생성
        self.figure = Figure()

        # Figure를 위한 캔버스 생성
        self.canvas = FigureCanvas(self.figure)

        # QWidget 위에 캔버스를 배치하기 위한 레이아웃
        layout = QVBoxLayout()
        layout.addWidget(self.canvas)

        # QWidget 생성 및 레이아웃 설정
        widget = QWidget(self)
        widget.setLayout(layout)
        self.setCentralWidget(widget)

        # 데이터 생성 및 차트 그리기
        self.draw_chart()

    def draw_chart(self):
        # Figure에서 Axes 객체 생성
        axes = self.figure.add_subplot(111)

        # 데이터 생성
        data = [1, 2, 3, 4, 5]

        # 차트 그리기
        axes.plot(data)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.show()
    sys.exit(app.exec_())

위 코드에서는 PyQt의 QMainWindow를 상속받은 MainWindow 클래스를 정의합니다. MainWindow 클래스에서는 Matplotlib의 Figure와 FigureCanvas를 생성한 뒤, QWidget 위에 캔버스를 배치하는 레이아웃을 설정합니다. 그리고 데이터를 생성하여 차트를 그리는 draw_chart() 메소드를 정의합니다.

seaborn을 이용한 데이터 시각화

seaborn은 Matplotlib를 기반으로 한 고급 데이터 시각화 라이브러리로, Matplotlib보다 더 간단하고 아름다운 시각화를 제공합니다. PyQt에서 seaborn을 사용하려면 다음과 같이 설치해야 합니다.

pip install seaborn

다음은 seaborn을 사용하여 데이터를 시각화하는 예제 코드입니다.

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget
import seaborn as sns

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # QWidget 위에 시각화를 보여줄 캔버스 생성
        self.canvas = QWidget()

        # seaborn을 이용한 시각화
        self.draw_chart()

        # 캔버스를 QMainWindow에 배치
        layout = QVBoxLayout()
        layout.addWidget(self.canvas)

        widget = QWidget(self)
        widget.setLayout(layout)
        self.setCentralWidget(widget)

    def draw_chart(self):
        # seaborn으로 데이터 시각화
        data = [1, 2, 3, 4, 5]
        sns.lineplot(data=data, ax=self.canvas)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.show()
    sys.exit(app.exec_())

위 코드에서는 PyQt의 QMainWindow를 상속받은 MainWindow 클래스를 정의합니다. MainWindow 클래스에서는 seaborn의 lineplot() 함수를 사용하여 데이터를 시각화합니다.

정리

PyQt를 사용하면 Matplotlib이나 seaborn과 같은 데이터 시각화 라이브러리를 쉽게 활용할 수 있습니다. 이러한 기능들을 활용하여 파이썬 프로그램에서 데이터 시각화를 간편하게 진행할 수 있습니다.