如何在 Matplotlib 中使用文本注释热图？

若要在 matplotlib 中使用文本注释热图，我们可以采取以下步骤 -

  • 使用4×4维度数组创建随机数据。

  • 使用 pcolor() 方法创建一个带非规则矩形网格的伪彩图。

  • 若要在像素中放置文本，我们可以使用 text() 方法。

  • 要显示图形，请使用show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(4, 4)
heatmap = plt.pcolor(data, cmap="PuBuGn_r")
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.show()

输出

更新时间：08-05-2021

1000+ 浏览

开启你的职业生涯

完成课程获得认证

开始
广告