如何将基于ISLR库的R线性回归代码改写为Python代码？

一、R代码转Python实现 #

原R代码的功能是加载ISLR包中的Auto数据集，构建mpg对horsepower的线性回归模型并输出模型摘要。用Python实现的话，推荐使用statsmodels库（输出的模型摘要和R的summary()格式接近），代码如下：

# 导入所需库
import statsmodels.api as sm
from statsmodels.formula.api import ols

# 获取Auto数据集（从ISLR的R数据集中加载）
auto_data = sm.datasets.get_rdataset("Auto", "ISLR").data

# 拟合线性回归模型
fit = ols("mpg ~ horsepower", data=auto_data).fit()

# 打印模型摘要
print(fit.summary())

如果你的环境中没有statsmodels，可以先通过pip安装：

pip install statsmodels

二、Jupyter Notebook运行R代码出错的常见排查点 #

你在Jupyter中运行R代码报错，通常是以下几个原因：

• 未安装R内核：Jupyter默认只支持Python，要运行R代码需要先安装IRkernel。打开R控制台，执行以下命令：

  install.packages("IRkernel")
  IRkernel::installspec()
  

  安装完成后重启Jupyter，就能选择R内核创建 notebook。
• 未安装ISLR包：R代码中用到的ISLR包需要提前安装，在R控制台执行：

  install.packages("ISLR")
  

  再运行library(ISLR)就不会报错。
• 代码拼写错误：检查library(ISLR)、data(Auto)等命令是否有拼写错误，比如大小写、单词拼写。
• 内核环境异常：如果内核加载失败，尝试重启Jupyter notebook，或者重新安装R内核。

内容的提问来源于stack exchange，提问作者Tahirie Coleman