您可以使用 seaborn regplot 制作回歸不連續點 Plot 嗎？

Question

有沒有辦法在 seaborn regplot 中包含變量d ，以便顯示 x=6 處的不連續性？ 如果不是，是否有替代 seaborn 的 regplot 的方法來創建經典回歸不連續點 plot？

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 )

Answer 1

是的，但您需要使用lmplot ，這令人討厭地需要 dataframe 輸入：

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)