您可以使用 seaborn regplot 制作回归不连续图吗?

Can you make a Regression Discontinuity Plot using seaborn regplot?

有没有办法在 seaborn regplot 中包含变量 d 以便显示 x=6 处的不连续性?如果没有,是否有替代 seaborn 的 regplot 来创建经典回归不连续图的方法?

import seaborn as sns
import numpy as np

x = np.array([2,4,5.99,6,8,10,12])
d = (x>=6)*1
y = 2*x + 10*d

sns.regplot(x = x, y = y, ci = None )

是的,但你需要使用 lmplot,这令人讨厌地需要数据帧输入:

import seaborn as sns
import numpy as np
import pandas as pd

x = np.array([2, 4, 5.99, 6, 8, 10, 12])
d = x >= 6
y = 2 * x + 10 * d
df = pd.DataFrame({"x": x, "y": y, "d": d})

sns.lmplot(data=df, x="x", y="y", hue="d", ci=None)