Python 热分布图

Question

我一直在尝试模拟空间中的流量分布，但无法弄清楚如何在 python 中使用 matplotlib、seaborn、plotly 等正确地做到这一点。

我有 3 个变量

X : 0 到 0.4 （网格到 142 片）

Y : 0 到 0.45（网格到 17767 个）

T : 值从 300 开尔文开始，沿房间分布 (142x17767)

为了解释更多，我有一个 142x17767 的矩形区域，我想在每个点上绘制热分布。我已经从 matplotlib 中尝试过 seaborn 的热图和 hist2d，但是这些方法要求 x 和 y 维度具有相同的长度。

Answer 1

我认为你需要的是pcolormesh。这是一个示例答案。您可以将下面的T 替换为您的实际 T 值。您收到错误的原因是因为在这样的 2d 热图上，您需要创建 x 和 y 点的网格。

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots(figsize=(6,6))

X = np.linspace(0, 0.4, 142)
Y = np.linspace(0, 0.45, 17767)

Xmesh, Ymesh = np.meshgrid(X,Y)

T = np.random.randint(300, 1000, Xmesh.shape)

plt.pcolormesh(X, Y, T)
plt.colorbar()
plt.show()