如何在R中将分组线性回归的R Squared与对应组名共同存入数据框

解决方法 #

你提取到的r_squareds向量本身就自带了对应的城市名作为向量名，直接提取该属性即可构造目标数据框，参考代码如下：

# 方法1：直接从R方向量提取对应城市名，适配性最高
r_squared_df <- data.frame(
  City = names(r_squareds),
  `R-Squared` = unname(r_squareds),
  check.names = FALSE
)

如果遇到向量名意外丢失的情况，也可以从原始数据集提取分组的城市名构造：

# 方法2：从原始数据提取城市名
r_squared_df <- data.frame(
  City = unique(df$City),
  `R-Squared` = r_squareds,
  check.names = FALSE
)

说明

• lmList按City分组拟合模型后，返回的汇总结果中r.squared向量的命名默认对应分组变量City的唯一值，和拟合的分组顺序完全匹配，不会出现对应错位的问题，优先选择方法1
• 代码中check.names = FALSE是为了保留R-Squared的列名格式，避免R自动将列名里的特殊符号替换为点

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