遍历所有列并在回归模型中添加所有可能的组合 - Python

Question

我想写一个函数，当我们传递一个包含A，B，C和Y列的数据帧时，该函数将返回一个模型，其中包含A，B，C的所有可能组合以及交互项。

我想要的输出是

的输出

smf.ols('Y~ A + B + C + A*B + B*C + C*A + A*B*C' , data = dataframe).fit()

最初我尝试过，

for column in df:
    print(df[column])

但我不知道如何继续这样做。这可能是相当开放的。但任何人都可以提出一个想法，我可以尝试并形成它。有人可以帮我这么做吗？

Answer 1

我不熟悉您正在处理的域名，但如果您希望A，B和C的所有组合一次至少两个，您可以获得他们喜欢这样

import itertools

itertools.chain(itertools.combinations(['A','B','C'], 2), itertools.combinations(['A', 'B', 'C,'], 1))

然后，您可以将其按到您想要的表达式中。