草庐IT

univariate

全部标签

python Pandas : how to run multiple univariate regression by group

假设我有一个DataFrame,其中有一列y变量和许多列x变量。我希望能够运行y与x1、y与x2的多个单变量回归,...,等等,并将预测存储回DataFrame。我还需要通过组变量来执行此操作。importstatsmodels.apiassmimportpandasaspddf=pd.DataFrame({'y':np.random.randn(20),'x1':np.random.randn(20),'x2':np.random.randn(20),'grp':['a','b']*10})defols_res(x,y):returnsm.OLS(y,x).fit().predict