python - 线性模型的R预测函数与Python等效吗？

线性模型的R预测函数与Python等效吗？

我确定scipy中有些东西可以在这里提供帮助，但是是否有等效的功能？

https://stat.ethz.ch/R-manual/R-patched/library/stats/html/predict.lm.html

最佳答案

Scipy具有大量具有预测方法的回归工具。尽管是IMO，但Pandas是最接近复制R功能的python库，并带有预测方法。 R和python中的以下代码段演示了相似之处。

R线性回归：

data(trees)
linmodel <- lm(Volume~., data = trees[1:20,])
linpred <- predict(linmodel, trees[21:31,])
plot(linpred, trees$Volume[21:31])

使用pandas ols在python中设置了相同的数据：

import pandas as pd
from pandas.stats.api import ols
import matplotlib.pyplot as plt

trees = pd.read_csv('trees.csv')
linmodel = ols(y = trees['Volume'][0:20], x = trees[['Girth', 'Height']][0:20])
linpred = linmodel.predict(x = trees[['Girth', 'Height']][20:31])
plt.scatter(linpred,trees['Volume'][20:31])