如何简化R语言中批量写入CSV文件的重复代码?
解决方案
你完全可以用lapply或者更方便的mapply/purrr::imap来简化代码,以下是几种可行的实现方式:
基础R实现(无需额外包)
利用mapply同时遍历列表元素和对应的名称,自动生成文件名并导出:
mapply(function(data, elem_name) { # 构造文件名:Weekly_ + 列表元素名 + .csv output_file <- paste0("Weekly_", elem_name, ".csv") # 导出CSV,保留原列名,不输出行名 write.csv(data, output_file, row.names = FALSE) }, weekly_data, names(weekly_data))
或者用lapply结合列表名称索引:
lapply(names(weekly_data), function(elem_name) { data <- weekly_data[[elem_name]] output_file <- paste0("Weekly_", elem_name, ".csv") write.csv(data, output_file, row.names = FALSE) })
tidyverse 风格实现(需安装purrr包)
如果习惯tidyverse语法,purrr::imap可以更直观地同时获取列表元素和名称:
library(purrr) imap(weekly_data, function(data, elem_name) { output_file <- paste0("Weekly_", elem_name, ".csv") write.csv(data, output_file, row.names = FALSE) }) # 更简洁的单行写法 imap(weekly_data, ~write.csv(.x, paste0("Weekly_", .y, ".csv"), row.names = FALSE))
注意事项
你原代码中NOx_2017_2021_all_site对应的导出语句漏加了.csv后缀,上述简化代码会自动统一添加后缀,避免这个问题。
内容的提问来源于stack exchange,提问作者Chauncey Tong
相关产品推荐
相关产品推荐

