如何通过循环或函数批量保存列表中按年份分类的DataFrame

问题说明 #

我有一个按年份分类存储DataFrame的列表mydata，创建代码如下：

mydata<-list()
mydata[["2010"]]<-data.frame(Age=c(12,13),Year=rep("2010",2))
mydata[["2011"]]<-data.frame(Age=c(14,15),Year=rep("2011",2))
mydata[["2012"]]<-data.frame(Age=c(16,17),Year=rep("2012",2))
mydata[["2013"]]<-data.frame(Age=c(18,17),Year=rep("2013",2))
mydata[["2014"]]<-data.frame(Age=c(16,27),Year=rep("2014",2))

我不想保存整个列表，而是要将每个年份的DataFrame单独存为带年份标识的RData文件。已经通过循环用assign批量创建了mydata_2010这类变量：

for (i in as.character(2010:2012)) {
  assign(paste0("mydata_",i), mydata[[i]])
}

但尝试直接在save中嵌套assign的代码未成功，需要解决批量保存的问题。

解决方案 #

方法一：直接操作列表（无需提前创建变量） #

最高效的方式是直接遍历列表元素，跳过全局环境赋值的步骤，直接保存：

# 保存所有年份的DataFrame
for (year_name in names(mydata)) {
  save(mydata[[year_name]], file = paste0("mydata_", year_name, ".RData"))
}

# 仅保存2010-2012的子集
target_years <- as.character(2010:2012)
for (year in target_years) {
  save(mydata[[year]], file = paste0("mydata_", year, ".RData"))
}

方法二：针对已创建的变量进行保存 #

如果已经通过assign在全局环境生成了mydata_2010这类变量，可以用get函数获取变量对象后保存：

for (i in as.character(2010:2012)) {
  var_name <- paste0("mydata_", i)
  save(get(var_name), file = paste0(var_name, ".RData"))
}

方法三：函数式编程风格（可选） #

如果习惯使用tidyverse工具，用purrr::iwalk可以更简洁地完成批量操作：

library(purrr)

# 保存所有年份
iwalk(mydata, ~save(.x, file = paste0("mydata_", .y, ".RData")))

# 仅保存2010-2012的子集
target_years <- as.character(2010:2012)
mydata[target_years] %>% 
  iwalk(~save(.x, file = paste0("mydata_", .y, ".RData")))

原代码失败原因 #

assign函数的返回值是被赋值的对象本身，但save函数需要接收对象名称的符号而非对象内容。直接将assign的结果传给save，会导致保存的变量名称不是预期的mydata_2010，而是临时的默认名称，因此无法得到正确命名的RData文件。

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