使用 matplotlib 的 2D 热图

Question

我是 python 新手，请多多包涵：

我一直在尝试绘制 2D 热图，类似于此处显示的：

http://mips.helmholtz-muenchen.de/plant/static/images/A_thal_LTRs.png

使用 contourf 或 colorbar 类，但它似乎不起作用。

我使用两个非常简单的数据集，如代码所示： `

import numpy as np
import matplotlib.pyplot as plt

abundance = [0.2,0.3,0.25,0.05,0.05,0.04,0.06]

grain_size = [200,100,70,50,10,5,1]

`

我希望 grain_size 数组成为我的 x_axis（在对数刻度上），我的颜色代表每个 grain_size 对应的丰度（所以 0.2 对应于 200，0.3 对应于 100 等等......）

所以我知道我需要标准化我的丰度数组以适应颜色条，但是然后呢？

非常感谢！

Answer 1

这是你想要的吗？

import matplotlib.cm as cm

ab = np.array(abundance)
gs = np.array(grain_size)
ab_norm = ab/ab.max()*100
plt.matshow([ab_norm], cmap=cm.gist_rainbow_r) 
plt.xticks(range(7), gs)
plt.yticks(range(1), ["abundance"])
plt.colorbar()
plt.show()

您可以通过选择另一个颜色图来更改颜色图，see here for some of them。

如果不是这样，如果你不明白，请告诉我。 希望这会有所帮助。