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

matplotlibserver side programmingprogramming

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

输出


相关文章