Python中空间数据的热图

Question

我正在写一个Python代码。 我的数据集由三列组成，第一列是坐标，第三列是热量估算

                   X              Y      heat
0      497935.000000  179719.000000  0.048428
1      497935.000000  179719.000000  0.029399
2      497935.000000  179719.000000  0.066734
3      497935.000000  179719.000000  0.065524
4      497935.000000  179719.000000  0.062458
5      497935.000000  179719.000000  0.032225
6      497935.000000  179719.000000  0.028557
7      497463.000000  179526.000000  0.000388
8      497899.000000  179305.000000  0.000733
9      497805.000000  179364.000000  0.000158
10     498264.000000  180349.000000  0.000036
11     498020.000000  179602.000000  0.003149
...              ...            ...       ...

我想在XY平面上绘制数据，为每个点指定一种颜色，并反映其热值。 目前我尝试了这个命令

plt.scatter（df.X，df.Y）

但在Python中，每个点的颜色都取决于它的热量。 提前致谢

Answer 1

从文档 ：

关键字c可以作为列的名称给出，以便为每个点提供颜色：

In [64]: df.plot.scatter(x='a', y='b', c='c', s=50);

因此，您需要做的是简单地指定heat柱包含有关每个点颜色的信息：

df.plot.scatter(x=data.X, y=data.Y, c=data.heat)

如果要应用自定义颜色贴图，还有cmap参数，允许您指定不同的颜色贴图

您还可以在scatter（）方法的文档中阅读更多相关信息。