如何在dplyr管道中进一步处理生成空数据的管道

时间:2015-01-13 02:04:05

标签: r dplyr

此问题发生在dplyr版本0.30。

我有一串%>%来自filter。有时,此过滤器会将数据框减少为无行。有些位于管道的下方,我有一个使用if来改变数据框的函数。但是,如果数据帧先前已减少到零行,则此函数会出错。

例如

data(mtcars)

stupid_function <- function(x){
    if( x == 6){
        return(2*x)
    } else {
        return(x)
    }
}

for(i in 6:10) {

    data <-
        mtcars %>% 
        filter(cyl == i) %>%
        rowwise() %>%
        mutate(carb2 = stupid_function(carb)) %>%
        group_by(carb2) %>%
        summarise(mean(wt))

    print(data)

}

适用于i = 6,但i = 7失败,例如

无论如何都要处理这个问题?我考虑过的两种方法是在中间拆分链,以便在过滤后检查数据是否有多行,或者将所有内容包装在tryCatch中。

1 个答案:

答案 0 :(得分:1)

首先,在最新版本的dplyr(0.4.0)中,过滤器不再崩溃,而是returns its input when the output is 0-sized(请参阅#782),因此您可能不会再出现错误。具体做法是:

library(dplyr)
data(mtcars)

stupid_function <- function(x){
  if(x == 6){
    return(2 * x)
  } else {
    return(x)
  }
}

for(i in 6:10) {

  data <-
    mtcars %>% 
    filter(cyl == i) %>%
    rowwise() %>%
    mutate(carb2 = stupid_function(carb)) %>%
    group_by(carb2) %>%
    summarise(mean(wt))

  print(data)

}

返回:

Source: local data frame [3 x 2]

  carb2 mean(wt)
1     1  3.33750
2     4  3.09375
3    12  2.77000
Source: local data frame [0 x 2]

Variables not shown: carb2 (dbl), mean(wt) (dbl)
Source: local data frame [4 x 2]

  carb2 mean(wt)
1     2 3.560000
2     3 3.860000
3     4 4.433167
4     8 3.570000
Source: local data frame [0 x 2]

Variables not shown: carb2 (dbl), mean(wt) (dbl)
Source: local data frame [0 x 2]

Variables not shown: carb2 (dbl), mean(wt) (dbl)
Warning messages:
1: Grouping rowwise data frame strips rowwise nature 
2: Grouping rowwise data frame strips rowwise nature 
3: Grouping rowwise data frame strips rowwise nature 
4: Grouping rowwise data frame strips rowwise nature 
5: Grouping rowwise data frame strips rowwise nature 

您可能还想在stupid_function中捕获0大小的输入,如下所示:

stupid_function <- function(x = NULL) {
  if (is.null(x)) {
    return(0)
  } else if(x == 6) {
    return(2 * x)
  } else {
    return(x)
  }
}

如果没有其他任何东西填充它,这会预先为x分配NULL并将0(或者您可以指定NULL)指定为返回。