Python：使用数据框进行颜色映射

Question

我有一个名为summed_data的数据框如下：

altitude    1000    2000    3000    4000
longitude
-159          0       0       0       1
-158          1       2       2       0

我正在尝试创建一个二维网格图，经度与海拔高度的关系，第三维（表0,0,0,1,1,2,2,0中的值）由颜色图表示为第三维尺寸。

我试过了：

plt.imshow(summed_data.values, cmap = 'RdBu')

这会产生：

但是，我希望色彩图与经度和海拔高度相关联。

修改

当我尝试以下答案时：

plt.imshow(summed_data.values, aspect = 'auto', extent = [-160,-150,0,5000],cmap = 'RdBu')

该数字与指定的经度和海拔高度无关：

框中只有经度= -158或-159的颜色以及相应的高度。

如何使用python中颜色贴图表示的summed_data表中的值绘制经度与海拔高度？