
一提到圖表,很多人可能會想到傳統(tǒng)的方式:條形圖、折線圖等。但今天我們要聊的是一種更高級的繪圖方式——矩陣顏色圖。它是通過一個二維的數(shù)據(jù)矩陣來生成彩色熱圖,用于展示數(shù)值分布或相關(guān)性分析。
在Python中,我們可以利用matplotlib
庫中的qcustomplot
插件輕松實現(xiàn)這一功能。首先,你需要安裝這個庫:
pip install qcustomplot
接著就可以開始繪制啦!代碼如下:
import numpy as np
from PyQt5.QtWidgets import QApplication, QMainWindow
from pyqtgraph.widgets.TreeWidget import TreeWidget
from pyqtgraph import PlotWidget, plot
import pyqtgraph as pg
app = QApplication([])
w = QMainWindow()
win = w centralWidget()
# 生成一個隨機矩陣,用于繪制顏色圖
matrix_data = np.random.rand(10, 10)
cp = pg.ColorMap(np.linspace(-5.0, 5.0), np.array([[255, 0, 0], [255, 165, 0], [255, 255, 0], [0, 255, 0], [0, 0, 255]]))
cp = pg.ColorMap(np.linspace(0.0, 1.0), np.array([[255, 0, 0], [255, 165, 0], [255, 255, 0], [0, 255, 0], [0, 0, 255]]))
p = win.addPlot()
p.setAspectLocked(True)
h = p.image(matrix_data, colorMap=cp)
在上面的代碼中,我們生成了一個隨機10x10矩陣,并用pyqtgraph
庫來繪制顏色圖。你可以根據(jù)自己的需求調(diào)整矩陣大小和顏色。