Return seaborn 热图中每个单元格的颜色

Question

我有一些数据正在以两种不同的方式可视化。第一个使用 seaborn.heatmap，第二个使用 seaborn.kdeplot。我想要的是使用热图中每个单元格的颜色为相应的 kdeplot 着色。示例：

this_matrix = np.random.normal(0,1,size=(100,100))
seaborn.heatmap(this_matrix)

我想要的是 return seaborn 在上述热图中的每个单元格中使用的颜色（rgb 元组）。

Answer 1

你可以像 seaborn 那样做：

import matplotlib.colors as mcolors
import numpy as np
import matplotlib.cm as cm

data = np.random.randn(5, 5)
norm = mcolors.Normalize()
data_normed = norm(data)
data_rgba = cm.viridis(data_normed)

请注意，如果您使用不同的颜色图，则应更改 viridis，如果指定 vmin 和 vmax，则应将它们作为参数传递给 Normalize

Return seaborn 热图中每个单元格的颜色

Return the color of each cell in a seaborn heatmap

python

visualization

matplotlib

seaborn