python - 如何在 seaborn pairplot 上绘制身份线？

Question

我正在使用 Seaborn 的配对图:

g = sns.pairplot(df)

是否可以在每个散点图上绘制标识线？

Answer 1

定义一个函数，它将在当前轴上绘制标识线，并使用 PairGrid.map_offdiag() 方法将其应用于网格的非对角线轴。

例如:

import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt

def plot_unity(xdata, ydata, **kwargs):
    mn = min(xdata.min(), ydata.min())
    mx = max(xdata.max(), ydata.max())
    points = np.linspace(mn, mx, 100)
    plt.gca().plot(points, points, color='k', marker=None,
            linestyle='--', linewidth=1.0)

ds = sns.load_dataset('iris')
grid = sns.pairplot(ds)
grid.map_offdiag(plot_unity)

这会在我的设置中生成以下图。您可以调整 plot_unity 函数的 kwargs 以根据需要设置绘图样式。