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 将字符串更改为 formatted string。
{} 内的部分采用 {value:format} 形式或只是 {value}（获取默认格式），并将被格式化文本替换。
格式规范是完整 mini-language 的一部分，其中 .2g 表示具有 2 位有效数字的“通用”格式。

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

1 个答案: