使用matplot lib制作2D热力图

Question

使用matplot lib制作2D热力图

pythonmatplotlib

3

我是新手，正在学习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轴（采用对数比例），颜色代表每个grain_size相应的丰度（因此0.2对应于200，0.3对应于100等等......）。

所以我知道我需要将丰度数组标准化以适合色条，但接下来怎么做呢？

非常感谢！

- Nimrodshn

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- jrjc · Accepted Answer

这是你想要的吗？

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()

你可以选择这里的其他调色板来更改调色板。

如果不是这个问题，或者有什么不理解的，请告诉我。希望能对你有所帮助。