python - 如何从图像上具有不同值的不同像素的坐标数据框创建热图

Question

我有一个来自 pandas 的数据框，例如:

在值行中可以找到 0 到 400 之间的数字。 我的想法是根据这个数据框在图像上创建一个热图。数据帧的这个值是同一张照片的不同像素变化。这可能吗？

我试过使用:

print(sns.heatmap(df))

作为数据框的 df。

我无法获取热图，所以我没有尝试将其等 ionic 化为图像。谢谢!!

Answer 1

我从数据框中获取列行:

df[["x", "y"]] = pd.DataFrame(df.Coordinates.to_list())

然后我用 mumpy 创建一个二维数组，其中所有元素的初始值为 0。数组的大小取决于“x”和“y”行的最大值:

import numpy as np
matrix = np.zeros((df.x.max()+1, df.y.max()+1))

然后我们使用 [x,y] 坐标和“值”行填充数组元素:

matrix[df.x, df.y] = df.Value

然后我们使用 seaborn 创建热图:

import seaborn as sns
sns.heatmap(matrix)