散点图和主题

This commit is contained in:
Irony 2019-10-02 23:03:59 +08:00
parent b8fe2b9727
commit 0a71f456dc
6 changed files with 454 additions and 0 deletions

362
QChart/ChartThemes.py Normal file
View file

@ -0,0 +1,362 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 2019/10/2
@author: Irony
@site: https://pyqt5.com , https://github.com/892768447
@email: 892768447@qq.com
@file: ChartThemes
@description: 图表主题动画等
"""
#############################################################################
##
## Copyright (C) 2013 Riverbank Computing Limited
## Copyright (C) 2012 Digia Plc
## All rights reserved.
##
## This file is part of the PyQtChart examples.
##
## $QT_BEGIN_LICENSE$
## Licensees holding valid Qt Commercial licenses may use this file in
## accordance with the Qt Commercial License Agreement provided with the
## Software or, alternatively, in accordance with the terms contained in
## a written agreement between you and Digia.
## $QT_END_LICENSE$
##
#############################################################################
import random
from PyQt5.QtChart import (QAreaSeries, QBarSet, QChart, QChartView,
QLineSeries, QPieSeries, QScatterSeries, QSplineSeries,
QStackedBarSeries)
from PyQt5.QtCore import pyqtSlot, QPointF, Qt
from PyQt5.QtGui import QColor, QPainter, QPalette
from PyQt5.QtWidgets import (QCheckBox, QComboBox, QGridLayout, QHBoxLayout,
QLabel, QSizePolicy, QWidget)
class ThemeWidget(QWidget):
def __init__(self, parent=None):
super(ThemeWidget, self).__init__(parent)
self.m_charts = []
self.m_listCount = 3
self.m_valueMax = 10
self.m_valueCount = 7
self.m_dataTable = self.generateRandomData(self.m_listCount,
self.m_valueMax, self.m_valueCount)
self.m_themeComboBox = self.createThemeBox()
self.m_antialiasCheckBox = QCheckBox("Anti-aliasing")
self.m_animatedComboBox = self.createAnimationBox()
self.m_legendComboBox = self.createLegendBox()
self.connectSignals()
# Create the layout.
baseLayout = QGridLayout()
settingsLayout = QHBoxLayout()
settingsLayout.addWidget(QLabel("Theme:"))
settingsLayout.addWidget(self.m_themeComboBox)
settingsLayout.addWidget(QLabel("Animation:"))
settingsLayout.addWidget(self.m_animatedComboBox)
settingsLayout.addWidget(QLabel("Legend:"))
settingsLayout.addWidget(self.m_legendComboBox)
settingsLayout.addWidget(self.m_antialiasCheckBox)
settingsLayout.addStretch()
baseLayout.addLayout(settingsLayout, 0, 0, 1, 3)
# Create the charts.
chartView = QChartView(self.createAreaChart())
baseLayout.addWidget(chartView, 1, 0)
self.m_charts.append(chartView)
chartView = QChartView(self.createBarChart(self.m_valueCount))
baseLayout.addWidget(chartView, 1, 1)
self.m_charts.append(chartView)
chartView = QChartView(self.createLineChart())
baseLayout.addWidget(chartView, 1, 2)
self.m_charts.append(chartView)
chartView = QChartView(self.createPieChart())
# Funny things happen if the pie slice labels no not fit the screen...
chartView.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
baseLayout.addWidget(chartView, 2, 0)
self.m_charts.append(chartView)
chartView = QChartView(self.createSplineChart())
baseLayout.addWidget(chartView, 2, 1)
self.m_charts.append(chartView)
chartView = QChartView(self.createScatterChart())
baseLayout.addWidget(chartView, 2, 2)
self.m_charts.append(chartView)
self.setLayout(baseLayout)
# Set the defaults.
self.m_antialiasCheckBox.setChecked(True)
self.updateUI()
def connectSignals(self):
self.m_themeComboBox.currentIndexChanged.connect(self.updateUI)
self.m_antialiasCheckBox.toggled.connect(self.updateUI)
self.m_animatedComboBox.currentIndexChanged.connect(self.updateUI)
self.m_legendComboBox.currentIndexChanged.connect(self.updateUI)
def generateRandomData(self, listCount, valueMax, valueCount):
random.seed()
dataTable = []
for i in range(listCount):
dataList = []
yValue = 0.0
f_valueCount = float(valueCount)
for j in range(valueCount):
yValue += random.uniform(0, valueMax) / f_valueCount
value = QPointF(
j + random.random() * self.m_valueMax / f_valueCount,
yValue)
label = "Slice " + str(i) + ":" + str(j)
dataList.append((value, label))
dataTable.append(dataList)
return dataTable
def createThemeBox(self):
themeComboBox = QComboBox()
themeComboBox.addItem("Light", QChart.ChartThemeLight)
themeComboBox.addItem("Blue Cerulean", QChart.ChartThemeBlueCerulean)
themeComboBox.addItem("Dark", QChart.ChartThemeDark)
themeComboBox.addItem("Brown Sand", QChart.ChartThemeBrownSand)
themeComboBox.addItem("Blue NCS", QChart.ChartThemeBlueNcs)
themeComboBox.addItem("High Contrast", QChart.ChartThemeHighContrast)
themeComboBox.addItem("Blue Icy", QChart.ChartThemeBlueIcy)
return themeComboBox
def createAnimationBox(self):
animationComboBox = QComboBox()
animationComboBox.addItem("No Animations", QChart.NoAnimation)
animationComboBox.addItem("GridAxis Animations", QChart.GridAxisAnimations)
animationComboBox.addItem("Series Animations", QChart.SeriesAnimations)
animationComboBox.addItem("All Animations", QChart.AllAnimations)
return animationComboBox
def createLegendBox(self):
legendComboBox = QComboBox()
legendComboBox.addItem("No Legend ", 0)
legendComboBox.addItem("Legend Top", Qt.AlignTop)
legendComboBox.addItem("Legend Bottom", Qt.AlignBottom)
legendComboBox.addItem("Legend Left", Qt.AlignLeft)
legendComboBox.addItem("Legend Right", Qt.AlignRight)
return legendComboBox
def createAreaChart(self):
chart = QChart()
chart.setTitle("Area chart")
# The lower series is initialized to zero values.
lowerSeries = None
y_points = []
for i, data_list in enumerate(self.m_dataTable):
upperSeries = QLineSeries(chart)
for j, (value, _) in enumerate(data_list):
y = value.y()
if lowerSeries is None:
upperSeries.append(QPointF(j, y))
y_points.append(y)
else:
new_y = y_points[i] + y
upperSeries.append(QPointF(j, new_y))
y_points[j] += new_y
area = QAreaSeries(upperSeries, lowerSeries)
area.setName("Series " + str(i))
chart.addSeries(area)
lowerSeries = upperSeries
chart.createDefaultAxes()
return chart
def createBarChart(self, valueCount):
chart = QChart()
chart.setTitle("Bar chart")
series = QStackedBarSeries(chart)
for i, data_list in enumerate(self.m_dataTable):
set = QBarSet("Bar set " + str(i))
for value, _ in data_list:
set << value.y()
series.append(set)
chart.addSeries(series)
chart.createDefaultAxes()
return chart
def createLineChart(self):
chart = QChart()
chart.setTitle("Line chart")
for i, data_list in enumerate(self.m_dataTable):
series = QLineSeries(chart)
for value, _ in data_list:
series.append(value)
series.setName("Series " + str(i))
chart.addSeries(series)
chart.createDefaultAxes()
return chart
def createPieChart(self):
chart = QChart()
chart.setTitle("Pie chart")
pieSize = 1.0 / len(self.m_dataTable)
for i, data_list in enumerate(self.m_dataTable):
series = QPieSeries(chart)
for value, label in data_list:
slice = series.append(label, value.y())
if len(series) == 1:
slice.setLabelVisible()
slice.setExploded()
hPos = (pieSize / 2) + (i / float(len(self.m_dataTable)))
series.setPieSize(pieSize)
series.setHorizontalPosition(hPos)
series.setVerticalPosition(0.5)
chart.addSeries(series)
return chart
def createSplineChart(self):
chart = QChart()
chart.setTitle("Spline chart")
for i, data_list in enumerate(self.m_dataTable):
series = QSplineSeries(chart)
for value, _ in data_list:
series.append(value)
series.setName("Series " + str(i))
chart.addSeries(series)
chart.createDefaultAxes()
return chart
def createScatterChart(self):
chart = QChart()
chart.setTitle("Scatter chart")
for i, data_list in enumerate(self.m_dataTable):
series = QScatterSeries(chart)
for value, _ in data_list:
series.append(value)
series.setName("Series " + str(i))
chart.addSeries(series)
chart.createDefaultAxes()
return chart
@pyqtSlot()
def updateUI(self):
theme = self.m_themeComboBox.itemData(
self.m_themeComboBox.currentIndex())
if self.m_charts[0].chart().theme() != theme:
for chartView in self.m_charts:
chartView.chart().setTheme(theme)
pal = self.window().palette()
if theme == QChart.ChartThemeLight:
pal.setColor(QPalette.Window, QColor(0xf0f0f0))
pal.setColor(QPalette.WindowText, QColor(0x404044))
elif theme == QChart.ChartThemeDark:
pal.setColor(QPalette.Window, QColor(0x121218))
pal.setColor(QPalette.WindowText, QColor(0xd6d6d6))
elif theme == QChart.ChartThemeBlueCerulean:
pal.setColor(QPalette.Window, QColor(0x40434a))
pal.setColor(QPalette.WindowText, QColor(0xd6d6d6))
elif theme == QChart.ChartThemeBrownSand:
pal.setColor(QPalette.Window, QColor(0x9e8965))
pal.setColor(QPalette.WindowText, QColor(0x404044))
elif theme == QChart.ChartThemeBlueNcs:
pal.setColor(QPalette.Window, QColor(0x018bba))
pal.setColor(QPalette.WindowText, QColor(0x404044))
elif theme == QChart.ChartThemeHighContrast:
pal.setColor(QPalette.Window, QColor(0xffab03))
pal.setColor(QPalette.WindowText, QColor(0x181818))
elif theme == QChart.ChartThemeBlueIcy:
pal.setColor(QPalette.Window, QColor(0xcee7f0))
pal.setColor(QPalette.WindowText, QColor(0x404044))
else:
pal.setColor(QPalette.Window, QColor(0xf0f0f0))
pal.setColor(QPalette.WindowText, QColor(0x404044))
self.window().setPalette(pal)
checked = self.m_antialiasCheckBox.isChecked()
for chartView in self.m_charts:
chartView.setRenderHint(QPainter.Antialiasing, checked)
options = QChart.AnimationOptions(
self.m_animatedComboBox.itemData(
self.m_animatedComboBox.currentIndex()))
if self.m_charts[0].chart().animationOptions() != options:
for chartView in self.m_charts:
chartView.chart().setAnimationOptions(options)
alignment = self.m_legendComboBox.itemData(
self.m_legendComboBox.currentIndex())
for chartView in self.m_charts:
legend = chartView.chart().legend()
if alignment == 0:
legend.hide()
else:
legend.setAlignment(Qt.Alignment(alignment))
legend.show()
if __name__ == '__main__':
import sys
from PyQt5.QtWidgets import QApplication, QMainWindow
app = QApplication(sys.argv)
window = QMainWindow()
widget = ThemeWidget()
window.setCentralWidget(widget)
window.resize(900, 600)
window.show()
sys.exit(app.exec_())

View file

@ -14,6 +14,8 @@
- [百分比柱状图表](#11百分比柱状图表)
- [横向柱状图表](#12横向柱状图表)
- [横向百分比柱状图表](#13横向百分比柱状图表)
- [散点图表](#14散点图表)
- [图表主题动画](#15图表主题动画)
## 1、折线图
[运行 LineChart.py](LineChart.py)
@ -83,3 +85,13 @@
[运行 HorizontalPercentBarChart.py](HorizontalPercentBarChart.py)
![HorizontalPercentBarChart](ScreenShot/HorizontalPercentBarChart.png)
## 14、散点图表
[运行 ScatterChart.py](ScatterChart.py)
![ScatterChart](ScreenShot/ScatterChart.png)
## 15、图表主题动画
[运行 ChartThemes.py](ChartThemes.py)
![ChartThemes](ScreenShot/ChartThemes.gif)

78
QChart/ScatterChart.py Normal file
View file

@ -0,0 +1,78 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 2019/10/2
@author: Irony
@site: https://pyqt5.com , https://github.com/892768447
@email: 892768447@qq.com
@file: ScatterChart
@description: 散点图表
"""
import random
from PyQt5.QtChart import QChartView, QChart, QScatterSeries
from PyQt5.QtCore import QPointF
from PyQt5.QtGui import QPainter
class Window(QChartView):
def __init__(self, *args, **kwargs):
super(Window, self).__init__(*args, **kwargs)
self.resize(400, 300)
# 抗锯齿
self.setRenderHint(QPainter.Antialiasing)
# 生成模拟数据
self.m_dataTable = self.generateRandomData(3, 10, 7)
# 图表
chart = QChart()
self.setChart(chart)
# 设置标题
chart.setTitle('Scatter chart')
# 添加Series
self.getSeries(chart)
# 创建默认xy轴
chart.createDefaultAxes()
chart.legend().setVisible(False)
def getSeries(self, chart):
for i, data_list in enumerate(self.m_dataTable):
series = QScatterSeries(chart)
for value, _ in data_list:
series.append(value)
series.setName('Series ' + str(i))
chart.addSeries(series)
def generateRandomData(self, listCount, valueMax, valueCount):
# 生成模拟随机数据
random.seed()
dataTable = []
for i in range(listCount):
dataList = []
yValue = 0.0
f_valueCount = float(valueCount)
for j in range(valueCount):
yValue += random.uniform(0, valueMax) / f_valueCount
value = QPointF(j + random.random() * valueMax / f_valueCount, yValue)
label = 'Slice ' + str(i) + ':' + str(j)
dataList.append((value, label))
dataTable.append(dataList)
return dataTable
if __name__ == '__main__':
import sys
from PyQt5.QtWidgets import QApplication
app = QApplication(sys.argv)
w = Window()
w.show()
sys.exit(app.exec_())

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.4 MiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 14 KiB

View file

@ -165,6 +165,8 @@ https://pyqt5.com 社区是专门针对PyQt5学习和提升开设的博客网站
- [百分比柱状图表](QChart/PercentBarChart.py)
- [横向柱状图表](QChart/HorizontalBarChart.py)
- [横向百分比柱状图表](QChart/HorizontalPercentBarChart.py)
- [散点图表](QChart/ScatterChart.py)
- [图表主题动画](QChart/ChartThemes.py)
- [PyQtGraph](PyQtGraph)
- [鼠标获取X轴坐标](PyQtGraph/mouseFlow.py)