更改 Seaborn 熱圖中缺失值的顏色

Question

考慮Seaborn 文檔中缺失值的示例：

corr = np.corrcoef(np.random.randn(10, 200))
mask = np.zeros_like(corr)
mask[np.triu_indices_from(mask)] = True
sns.heatmap(corr, mask=mask, vmax=.3, square=True)

如何將缺失值的顏色更改為例如黑色？ 缺失值的顏色應獨立於熱圖的配色方案指定，它可能不存在於配色方案中。

我嘗試添加facecolor = 'black'但這沒有用。 顏色可能會受到例如sns.axes_style("white")但我不清楚如何使用它來設置任意顏色。

Answer 1

您可以使用以下代碼：

corr = np.corrcoef(np.random.randn(10, 200))
mask = np.zeros_like(corr)
mask[np.triu_indices_from(mask)] = True
g = sns.heatmap(corr, mask=mask, vmax=.3, square=True)
g.set_facecolor('xkcd:salmon')

您需要在繪圖對象上使用set_facecolor 。 更改為您想要的任何顏色。

導致此圖：

Answer 2

另一種選擇是使用 sns.set_style() 在 seaborn 中設置活動樣式參數：

sns.set_style("white",  {'figure.facecolor': 'black'})
corr = np.corrcoef(np.random.randn(10, 200))
mask = np.zeros_like(corr)
mask[np.triu_indices_from(mask)] = True
sns.heatmap(corr, mask=mask, vmax=.3, square=True)
plt.axis("off")
plt.show()

完整的參數列表可以在這里找到： https : //seaborn.pydata.org/tutorial/aesthetics.html#overriding-elements-of-the-seaborn-styles