tidyr `fill()` 可以与 R 中的 if_else() 一起使用吗？答案

【问题标题】：Can tidyr `fill()` be used with if_else() in R?tidyr `fill()` 可以与 R 中的 if_else() 一起使用吗？
【发布时间】：2019-10-28 14:49:08
【问题描述】：

library(tidyverse)
df <- tibble(col1 = c("a", "a", "b", "b", "c", "c"),
             col2 = c(2, NA, 5, NA, 7, NA))
#> # A tibble: 6 x 2
#>   col1   col2
#>   <chr> <dbl>
#> 1 a         2
#> 2 a        NA
#> 3 b         5
#> 4 b        NA
#> 5 c         7
#> 6 c        NA

让我们从上面的数据框开始。我想填写col2，除非col1 中的值是a。解决方案如下所示：

#> # A tibble: 6 x 2
#>   col1   col2  col3
#>   <chr> <dbl> <dbl>
#> 1 a         2     2
#> 2 a        NA    NA
#> 3 b         5     5
#> 4 b        NA     5
#> 5 c         7     7
#> 6 c        NA     7

我在下面的尝试不起作用。如何让 tidyr::fill() 在这个 if_else() 上下文中工作？

df %>% mutate(col3 = if_else(col1 != "a", fill(col2), col2))
#> Error in UseMethod("fill_") : no applicable method for 'fill_' 
#> applied to an object of class "c('double', 'numeric')"

【问题讨论】：

标签： r if-statement tidyr

【解决方案1】：

我不认为这是该函数支持的行为。相反，您可以这样做：

df %>%
 filter(col1 == "a") %>%
 bind_rows(df %>%
            filter(col1 != "a") %>%
            fill(col2))

  col1   col2
  <chr> <dbl>
1 a         2
2 a        NA
3 b         5
4 b         5
5 c         7
6 c         7

或者如果你真的需要col3，那么你可以使用na.locf() from zoo:

df %>%
 mutate(col3 = ifelse(col1 != "a", na.locf(col2), col2))

  col1   col2  col3
  <chr> <dbl> <dbl>
1 a         2     2
2 a        NA    NA
3 b         5     5
4 b        NA     5
5 c         7     7
6 c        NA     7

【讨论】：