动态使用导入的 csv 文件名导出 csv 的 R 函数

Question

我正在尝试编写一个导入 .csv 文件的 R 函数，对其进行一些修改（此处删除空列）并将修改后的数据框导出为新的 .csv 文件。我有两个问题：

1. 我希望函数使用“primary.csv”文件的名称并向其添加“modified”，以便导出的 .csv 文件类似于“primary_modified.csv”

2. 我希望该函数作为一个循环工作，这样我就可以同时为多个数据帧运行它，而不是一个一个地运行。

感谢大家。

这是我所拥有的：

# create a function to remove all empty columns and save and write a new csv
removeEmptyCol <- function(x){
   df <- read.csv(x)
   colSums(is.na(df) | df == "")
   empty_columns <- colSums(is.na(df) | df == "") == nrow(df)
   df <- df[, !empty_columns]
   df[is.na(df)] <- ""
# need to change the file name for each dataframe 
   write.csv(df, "Trimmed-1.csv")

Answer 1

因为您没有提及其他地方的并行处理，我将您的问题解释为需要一个 for 循环来导入 .csv 文件，执行一个或多个操作，然后在导入另一个 .csv 文件并执行之前导出文件同一系列操作。

这是一个使用 tidyverse 的解决方案：

library(tidyverse)
  
remove_empty_col <- function (dir) {
  files <- list.files(path = dir, pattern = "*.csv", full.names = TRUE)
  
  for (file in files) {
    
    new_name <- str_replace(file, "\\.csv$", "_modified.csv") 
    
    df <- read_csv(file)
    
    df <- df %>%
       select_if(~!(all(is.na(.)) | all(. == "")))
    
    write_csv(df, paste0(new_name)) }
  
}

out <- remove_empty_col("/dir/path/to/data/sets")