使用matplotlib和numpy创建3列表的heatmap

Question

我有3个Python列表:x=[x0,x1,x2,...,xn]，y=[y0,y1,y2,...,yn]和v=[v0,v1,v2,...,vn]以及我需要通过创建一个heatmap来可视化数据，其中坐标(x[k],y[k])，值v[k]可视化，结果可能类似于GnuPlot heatmap xyz中的结果。由于系统的限制，我不能使用其他第三方工具，除了numpy和matplotlib。

我已经找到了一些相关的主题（用pcolor在matplotlib中生成Heatmap？，使用散点数据集在matplotlib中生成Heatmap）但似乎没有解决相同的问题。

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)