来自 csv 的 Python 热图

Question

我看过一些关于如何制作热图的示例，但我不止一次地将头撞在墙上，因为我真的不知道我的 .csv 文件必须是什么样子才能用于创建热图。我试图想象的看起来如下所示：

我见过 numpy 和 matplotlib 的解决方案看起来非常好，但我总是无法将 .csv 转换为可用格式。如果有人可以展示/告诉我在哪里/如何使其可用，那就太好了。

Answer 1

您可以轻松地使用 pandas 及其read_csv函数来读取文件，甚至使用 matplot 来绘制热图。

第一列（索引列）可能是给您带来麻烦的原因。您可以从数据中删除它（并使用 panda 的全新索引），或者在调用该函数时，将其用作 read_csv 函数中的 index_col 。

import pandas as pd
import matplotlib.pyplot as plt

path_to_csv= "insert here the path to your csv file"
df= pd.read_excel(path_to_csv ,index_col=0)

plt.imshow(df,cmap='hot',interpolation='nearest')

plt.show()

我在这里用你的数据尝试了这个脚本，它起作用了，

我希望它能帮助你！