在函数中跳过空数据帧的语句

Question

我在R 中有一个类似的函数，它使用df 作为主要参数。但是，df 可能为空，如果我使用 lapply 运行该函数，我想跳过这些 例如lapply(FileNames, function(y) my.function(y))

我的函数是这样的

my.function <- function(table) {
file <- read.csv(table, header=T)

  if(nrow(table) == 0) 
   {next}

#skip the df if is empty and stop processing that df
#do more stuffs if not empty
}

到目前为止我已经尝试过了，但是如果数据框为空，整个过程就会崩溃，有什么提示吗？

Answer 1

假设你有一个这样的数据框列表：

my_list <- list(data.frame(a = 1:3, b = 1:3),
                data.frame(a = c(), b = c()),
                data.frame(a = 1:5, b = 1:5))

my_list
#> [[1]]
#>   a b
#> 1 1 1
#> 2 2 2
#> 3 3 3
#> 
#> [[2]]
#> data frame with 0 columns and 0 rows
#> 
#> [[3]]
#>   a b
#> 1 1 1
#> 2 2 2
#> 3 3 3
#> 4 4 4
#> 5 5 5

那么你的函数可能是这样的：

my_func <- function(df) {

  # return NULL and stop function if dataframe has zero rows
  if(nrow(df) == 0) return(NULL)

  # Do work on dataframes with one or more rows
  data.frame(sum = df$a + df$b)
}

所以现在如果我们lapply 我们得到：

result <- lapply(my_list, my_func)

result
#> [[1]]
#>   sum
#> 1   2
#> 2   4
#> 3   6
#> 
#> [[2]]
#> NULL
#> 
#> [[3]]
#>   sum
#> 1   2
#> 2   4
#> 3   6
#> 4   8
#> 5  10

如果您想从列表中删除 NULL 值，您可以这样做：

result[!sapply(result, is.null)]
#> [[1]]
#>   sum
#> 1   2
#> 2   4
#> 3   6
#> 
#> [[2]]
#>   sum
#> 1   2
#> 2   4
#> 3   6
#> 4   8
#> 5  10

---

更新

如果遇到空文件，要在函数中包含警告消息，我们可以这样做：

my_func <- function(df) {
  
  # return NULL and stop function if dataframe has zero rows
  if(nrow(df) == 0) {
    warning("This file is empty")
    return(NULL)
  }
  
  # Do work on dataframes with one or more rows
  data.frame(sum = df$a + df$b)
}

^{由reprex package (v2.0.0) 于 2021 年 11 月 15 日创建}