python - 单元格之间没有线条的 Seaborn 热图

Question

我正在尝试使用带有透明色图的 seaborn 创建热图，因为图像应该显示在背景中。到目前为止，热图创建工作正常，但是即使热图的线宽设置为 0.0，单元格之间的一些线仍然可见。

创建热图的代码如下所示:

ax = sns.heatmap(image, cmap="rocket_r", linewidths=0.0)
ax.collections[0].set_alpha(0.5)

其中 image 是 64x64 numpy 数组。生成的热图如下所示: Heatmap (抱歉没有足够的声誉嵌入图片)

问题在于单元格之间的细线。奇怪的是，它们并不在每一个边缘。

有谁知道如何去除这些线条？

非常感谢

---

更新 1(完整的工作示例):

image = np.array([[1, 1, 2, 2], [3, 3, 3, 3], [4, 5, 4, 5], [6, 6, 6, 6]])
ax = sns.heatmap(image, cmap="rocket_r", linewidths=0.0)
ax.collections[0].set_alpha(0.5)
plt.show()

结果是这张热图:

在这里您可以看到每一列之间都有细线，但第一行和第二行之间没有任何线。

Answer 1

线条是半透明 block 的重叠，无法在像素网格上完美对齐。

阿尔法混合

一个选项是不使用透明度，而是使用 alpha 混合创建不透明的颜色。

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

def get_alpha_blend_cmap(cmap, alpha):
    cls = plt.get_cmap(cmap)(np.linspace(0,1,256))
    cls = (1-alpha) + alpha*cls
    return ListedColormap(cls)

image = np.array([[1, 1, 2, 2], [3, 3, 3, 3], [4, 5, 4, 5], [6, 6, 6, 6]])
ax = sns.heatmap(image, cmap=get_alpha_blend_cmap("rocket_r", 0.5), linewidths=0.0)

plt.show()

这样做的一个明显优势是颜色条与热图具有相同的颜色。

增加dpi

如果以上不是一个选项，您可以在保存时增加 dpi。

import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns

image = np.array([[1, 1, 2, 2], [3, 3, 3, 3], [4, 5, 4, 5], [6, 6, 6, 6]])
ax = sns.heatmap(image, cmap="rocket_r", linewidths=0.0, edgecolor="none", alpha=0.5)
plt.savefig("test.png", dpi=1000)

当然，这对屏幕上显示的数字没有任何影响。

即时显示

最后，考虑在这里不使用 seaborn，而是使用 matplotlib imshow 绘图。

import matplotlib.pyplot as plt
import seaborn as sns
plt.style.use("seaborn-dark")
plt.rcParams["axes.facecolor"] = "white"
import numpy as np

image = np.array([[1, 1, 2, 2], [3, 3, 3, 3], [4, 5, 4, 5], [6, 6, 6, 6]])
im = plt.imshow(image, cmap="rocket_r", alpha=0.5)
plt.colorbar(im)
plt.gca().set(xticks=(range(image.shape[1])),yticks=(range(image.shape[0])))
plt.show()