python - Seaborn 热图,单元格之间没有线条
问题描述
我正在尝试使用带有透明颜色图的 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()
结果是这个热图:
在这里,您可以看到每列之间有细线,但第一行和第二行之间没有任何线。
解决方案
这些线是半透明补丁的重叠,无法在像素网格上完美对齐。
阿尔法混合
一种选择是不使用透明度,而是使用 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,而是使用 matplotlibimshow
图。
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()
推荐阅读
- node.js - TypeError:无法读取未定义的属性“过滤器” - 不确定如何修复
- node.js - 将文件上传到在 AWS 上运行的节点/快递实例(不在本地 DEV 实例上)时遇到 1MB 限制
- java - 如果无法捕获或抛出它,如何处理由较低方法生成的通用异常
- excel - 使用 Excel VBA 获取某个日期的汇率
- java - tomcat不启动war应用
- python - 如何按单词字符串中的数字对列表进行排序?
- python - 如何使用原始字符串路径从 Pathlib 路径中提取父文件夹?
- python - 为什么 DQNAgent.fit 会向我的输入数据添加额外的维度?
- xcode - 在 mac big sur 上下载和更新 Xcode
- javascript - React Native useEffect - 警告:效果函数不得返回除用于清理的函数之外的任何内容