python - 带有 X、Y 数据的 Matplotlib 热图

Question

我想使用 pcolor 或其他热图库在 matplotlib 中制作热图。我找到了很多很好的例子，但无法确定如何以正确的格式获取我的数据，或者使用我的数据所在的格式进行绘图。

这是我的数据设置方式

X  Y  Value
0  1  .6
0  2  .3
0  3  .2
1  1  .8
1  2  .4
1  3  .9

因此，X 和 Y 列表示 (X,Y) 对，其中 Value 是值对应的单元格。我正在努力寻找一种方法来转换数据以使用 pcolor 或其他绘图方法。任何帮助将不胜感激。

Answer 1

看起来您使用的是 pandas 数据框。在将枢轴数据框绘制为表格并使用热图方法之前，即来自 seaborn:

import seaborn as sns 
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_clipboard()
table = df.pivot('Y', 'X', 'Value')
ax = sns.heatmap(table)
ax.invert_yaxis()
print(table)
plt.show()

输出:

X    0    1
Y
1  0.6  0.8
2  0.3  0.4
3  0.2  0.9