用Python库进行多元多重回归

Question

有没有Python中可执行多元多重回归（多重回归与多个因变量）的库？

问候并提前致谢。

- A. Esquivias

---

请求图书馆推荐明确不属于讨论范围。请参阅：[help/on-topic]，[tour]，[ask]。 - AMC

Answer 1

您可以尝试使用sklearn中的模块，响应变量可以是2维或更高维度的，我认为它适用于OLS（线性回归）、Lasso和Ridge。而statsmodels中的模型只能处理一个响应变量（我刚刚检查过）。

示例数据集：

import pandas as pd
from sklearn.datasets import load_iris
iris = load_iris()
df = pd.DataFrame(data= iris['data'],
                     columns= iris['feature_names'] )

df.shape
(150, 4)

现在我们进行适配：

from sklearn import linear_model
clf = linear_model.LinearRegression()
clf.fit(df[['sepal length (cm)']],df[['petal length (cm)','petal width (cm)']])
clf.coef_

array([[1.85843298],
       [0.75291757]])

您可以看到，在这种情况下，系数与拟合一个响应时相同：

clf.fit(df[['sepal length (cm)']],df[['petal width (cm)']])
clf.coef_
array([[0.75291757]])

- StupidWolf

---

询问图书馆推荐明确不属于讨论范围。请参阅：[help/on-topic]、[tour]、[ask]。 - AMC