如何用matplotlib中的文本注释热图？ plt.pcolor(rand(5,5)) 我怎么可以标注与实际数字热图绘制：

Question

我在matplotlib绘制使用热图？意思是在绘制的热图的每个单元格中，将对应于该单元格的值放入传递给pcolor的5x5矩阵中。谢谢。如何用matplotlib中的文本注释热图？</p> <p><code>plt.pcolor(rand(5,5))</code></p> <p>我怎么可以标注与实际数字热图绘制：

Answer 1

没有自动功能做这样的事情，但你可以通过每个点和循环把文本在适当的位置：

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

HTH

Answer 2

的seaborn热图通过设置annot=True自动完成这项工作。

查看this举例。