如何注释热图中的特定单元格？

Question

我怎样才能只注释四个角单元格？

for N in Nvals:
    for R in Rvals:
        filename = "data"
        filename += "-N" + str(N) + "-R0" + str(R)
        filename += ".txt"

        data = []
        with open(filename) as file:
            for line in file.readlines():
                #line = line.strip('"\'`(), False True \n')
                try:
                    value = float(line)
                except ValueError:
                    pass
                else:
                    data.append(value)

        data = np.array(data)
        mean.append(np.mean(data))
               
    mean = np.array(mean).reshape(11,10)

    plt.figure(1)
    plt.figure(figsize=(3,3), dpi=175)

    res=sns.heatmap(mean,square=True,
                norm=LogNorm(0.01, 100),
            annot=True,annot_kws={'size': 4},
            xticklabels=False, yticklabels=False,
            cbar = False,
            cmap="viridis_r")
    res.invert_yaxis()

Answer 1

Seaborn 的heatmap允许明确给出一个二维数组作为注释。 例如，您可以创建一个列表列表，其中包含除角落之外的所有空字符串：

from matplotlib import pyplot as plt
from matplotlib.colors import LogNorm
import numpy as np
import seaborn as sns

mean = 100 ** np.random.rand(11, 10)
plt.figure(1, figsize=(3, 3), dpi=175)

annot_values = np.full_like(mean, np.nan)
for i, j in [(0, 0), (0, -1), (-1, 0), (-1, -1)]:
    annot_values[i, j] = mean[i, j]
m, n = mean.shape
annot_values = [[f'{mean[i, j]:.2g}' if i in (0, m - 1) and j in (0, n - 1) else ''
                 for j in range(n)]
                for i in range(m)]
ax = sns.heatmap(mean, square=True,
                 norm=LogNorm(0.01, 100),
                 annot=annot_values, annot_kws={'size': 10}, fmt="s",
                 xticklabels=False, yticklabels=False,
                 cbar=False,
                 cmap="viridis_r")
ax.invert_yaxis()
plt.tight_layout()
plt.show()

PS：请注意，每次调用plt.figure()时都会创建一个新图形。 因此，建议将两个调用合二为一，防止创建空图。

关于f'{mean[i, j]:.2g}' ：

• f将字符串更改为格式化字符串。
• {}中的部分采用{value:format}或{value}形式（获取默认格式），并将替换为格式化文本。
• 格式规范是完整的mini-language的一部分，其中.2g表示具有2有效数字的“通用”格式。