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（）方法的文檔中閱讀更多相關信息。