python - 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 平面上绘制数据，为每个点分配一种颜色以反射(reflect)其热值。目前我尝试使用命令

plt.scatter(df.X, df.Y)

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

Answer 1

来自documentation :

The keyword c may be given as the name of a column to provide colors for each point:

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 参数，允许您指定不同的 color map

您还可以在 scatter() 的文档中阅读有关 in 的更多信息。方法。