python - Seaborn FacetGrid + 热图灰色边框

Question

给出以下示例:

A = [0, 1, 2]
B = [3, 4, 5]
C = [6, 7, 8, 9]

df = pd.DataFrame(data = {'D': np.random.randn(3*3*4)}, 
                  index = pd.MultiIndex.from_product([A, B, C], names=['A', 'B', 'C']))

df.reset_index(['B', 'C'], inplace=True)

def facet_heatmap(data, color):
    data = data.pivot(columns='B', values='D')
    ax = sns.heatmap(data, square=True, cmap="coolwarm", linewidths=0.0, rasterized=True, cbar=False)
    ax.invert_yaxis()

g = sns.FacetGrid(df, col='C', col_wrap=2)
g.map_dataframe(facet_heatmap)

当使用可选参数 square=True 时，网格的每个子图中都有一些灰色边框。但是，当 square=False (默认值)时，灰色边框消失。有人知道如何在保持 square=True 的同时删除灰色边框吗？

Answer 1

这不是边框，只是灰色背景。您可以使用seaborn的set_style函数将其设置为白色:

sns.set_style("whitegrid")