使用matplotlib和numpy創建3個列表的熱圖

Question

我有3個Python列表： x = [x0, x1, x2, ..., xn] ， y = [y0, y1, y2, ..., yn]和v = [v0, v1, v2, ..., vn]以及通過創建熱圖來可視化數據所需執行的操作，其中在坐標(x[k], y[k]) ，值v[k]可視化，結果可能類似於結果在GnuPlot熱圖XYZ中 。 由於系統限制，除numpy和matplotlib之外，我無法使用其他第三方工具。

我找到了一些相關的主題（ 使用pcolor在matplotlib中使用Heatmap ? ， 使用散點數據集在MatPlotLib中生成加熱圖 ），但似乎無法解決同一問題。

Answer 1

如果編碼的矩陣對於內存來說不是太大，則可以使用數組切片將其輕松轉換為密集的numpy數組：

import numpy as np
import matplotlib.pyplot as plt

x = [1, 0]
y = [0, 1]
v = [2, 3]

M = np.zeros((max(x) + 1, max(y) + 1))
M[x, y] = v

fig, ax = plt.subplots()
ax.matshow(M)