如何用matplotlib中的文本注释热图？

Question

我正在matplotlib中绘制热图，使用：

plt.pcolor(rand(5,5))

如何用实际绘制的数字标注热图？也就是说，在绘制的热图的每个单元中，将对应于该单元的值放入传递给pcolor的5x5矩阵中。谢谢。

Answer 1

seaborn heatmap通过设置annot=True自动执行该任务。

有关示例，请参见this。

Answer 2

没有自动功能来执行此操作，但您可以循环遍历每个点并将文本放在适当的位置：

import matplotlib.pyplot as plt
import numpy as np

data = np.random.rand(5, 4)
heatmap = plt.pcolor(data)

for y in range(data.shape[0]):
    for x in range(data.shape[1]):
        plt.text(x + 0.5, y + 0.5, '%.4f' % data[y, x],
                 horizontalalignment='center',
                 verticalalignment='center',
                 )

plt.colorbar(heatmap)

plt.show()

高温高压