使用 python ols 预测值

範例：

# -*- coding: utf-8 -*-import pandas as pdimport statsmodels.formula.api as smimport statsmodels.regression.linear_model as smimport statsmodels.api as sm2TV = [230.1, 44.5, 17.2, 151.5, 180.8]Radio = [37.8,39.3,45.9,41.3,10.8]Newspaper = [69.2,45.1,69.3,58.5,58.4]Sales = [22.1, 10.4, 9.3, 18.5,12.9]df = pd.DataFrame({'TV': TV,                    'Radio': Radio,                    'Newspaper': Newspaper,                    'Sales': Sales})Y = df.SalesX = df[['TV','Radio','Newspaper']]X = sm2.add_constant(X)model = sm.OLS(Y, X).fit()new_x = df.loc[df.Sales.notnull(), ['TV', 'Radio', 'Newspaper']].valuesnew_x = sm2.add_constant(new_x) # sm2 = statsmodels.apiy_predict = model.predict(new_x)print(df)print(y_predict)