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 数组。 生成的热图如下所示：热图（抱歉没有足够的声誉来嵌入图片）

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

任何人都知道如何摆脱这些线？

非常感谢

---

更新 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

这些线是半透明贴片的重叠，不能在像素网格上完美对齐。

alpha混合

一个选项是不使用透明度，而是使用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)

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

imshow

最后，考虑不在这里使用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()

Answer 2

我刚刚遇到这个问题。 我需要将 plot 上传到背面，所以我不喜欢 dpi 解决方案。 这是我根据 OP 的解决方案得出的结论：

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.1)
colors = ax.collections[0].get_facecolors()
ax.collections[0].set_edgecolors(colors)
plt.imshow()

这个想法是用单元格的面色在每个单元格周围创建一个细边。 这会在不更改原始单元格 colors 的情况下删除这些行。