QCustomPlot PyQt5 绘图代码示例

QCustomPlot PyQt5 绘图代码示例

from PyQt5.QtGui import QPen, QBrush, QColor
from QCustomPlot2 import *
from PyQt5.QtWidgets import QApplication
from PyQt5.QtCore import Qt
import sys
import math

app = QApplication(sys.argv)

# 创建绘图窗口
customPlot = QCustomPlot()
# 添加绘图图像
graph0 = customPlot.addGraph()
# 设置绘图颜色
graph0.setPen(QPen(Qt.blue))
# graph0.setBrush(QBrush(QColor(0, 0, 255, 20))) # brush可以填充曲线与横轴所夹的面积

graph1 = customPlot.addGraph()
graph1.setPen(QPen(Qt.red))
# 数据生成
x, y0, y1 = [], [], []
for i in range (251):
    x.append(i)
    y0.append(math.exp(-i/150.0)*math.cos(i/10.0)) # exponentially decaying cosine
    y1.append(math.exp(-i/150.0))                  # exponential envelope
# 添加数据
graph0.setData(x, y0)
graph1.setData(x, y1)
# 自动设置坐标轴范围及刻度
customPlot.rescaleAxes()
# 添加一些交互功能，拖拽图像，缩放，选择曲线
customPlot.setInteractions(QCP.Interactions(QCP.iRangeDrag | QCP.iRangeZoom | QCP.iSelectPlottables))
# 设置窗口大小
customPlot.setGeometry(400, 250, 542, 390)
# 显示图像
customPlot.show()

sys.exit(app.exec_())