Seaborn FacetGrid + 热图灰色边框

Seaborn FaceGrid + heatmap grey borders

给定以下示例:

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 的同时删除灰色边框?

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

sns.set_style("whitegrid")