散点图和主题

QChart/ChartThemes.py

@@ -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: [email protected]
+@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_())

QChart/README.md

@@ -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)