如何在matplotlib中用文本注釋熱圖

Question

我正在使用以下命令在 matplotlib 中繪制熱圖：

plt.pcolor(rand(5,5))

如何使用繪制的實際數字注釋熱圖？ 這意味着在繪制的熱圖的每個單元格中，將與該單元格對應的值放入傳遞給pcolor的 5x5 矩陣中。 謝謝。

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自動完成工作。

請參閱此示例。