在 Python 中绘制一行预测值是否可行?在 R 中,您可以使用 绘制预测值线lines(x$age, predict(result, data.frame(x$height, x$weight)))
。我想要做的是绘制从回归结果确定的这些预测值,但也想要绘制除了简单的单变量回归之外的任何东西,这可以在 NumPy 和 matplotlib 中完成。
例如,在以下代码中:
import statsmodels.formula.api as smf
df = pd.read_csv("sample.csv")
result = smf.ols("res~height+weight+I((180-height^2))+I((60-weight)^3)", data=df).fit()
我想使用 的回归结果smf.ols()
,这比通常的多元回归(比如res~height+weight
)要复杂一些。那么如何确定预测值并绘制这些值呢?
谢谢。