Python热分布图

时间:2019-02-22 18:28:41

标签: python plot heatmap distribution fluid-dynamics

我一直在尝试模拟空间中的流分布,但无法弄清楚如何使用matplotlib,seaborn,plotly等在python中正确地实现它。

我有3个变量

X:0至0.4(划分为142个)

Y:0至0.45(划分为17767个)

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

为进一步说明,我有一个142x17767的矩形区域,我想在每个点上绘制热量分布。我已经尝试过matplotlib中的seaborn的热图和hist2d,但是这些方法要求x和y维度具有相同的长度。

1 个答案:

答案 0 :(得分:0)

您需要的是我认为的pcolormesh。这是一个示例答案。您可以将下面的T替换为实际的T值。出现错误的原因是,在这样的二维热图上,您需要创建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()

enter image description here