动态case_when，允许不同数量的条件和条件本身

Question

我正在寻找一种动态的方法来指定一些“条件参数”，然后在操作时将其输入到一个case_（或者其他更适合该问题的方法）。

因此，我的目标是在调用时将条件的规范与case_分开，例如，用户可以在txt文件或R中的列表中键入条件，然后我会在调用时将该信息提供给case_（或任何其他函数，如果更合适的话）。

因此，假设下面的数据是我想要创建一个额外的变量来重新编码x，我可以这样做：

df <- data.frame(x = 1:10)

df |>
  mutate(x2 = case_when(x < 4 ~ 1,
                        x >= 4 & x <=7 ~ 2,
                        TRUE ~ 3))

现在，我想要实现的是，让这段代码更加灵活，我可以在外部指定case_when条件，然后重新编码。

例如，它可能看起来像：

all_conditions <- list(1 = "x < 2",
                       2 = "x >= 2 & x < 5",
                       3 = "x >= 5 & x < 9",
                       4 = "TRUE")

然后我可以做一些：

df |>
  mutate(x2 = do(case_when, all_conditions))

编辑：虽然示例显示了@Mael解决方案适用的数值类型变量，但该解决方案也适用于条件可能类似于“x==”abc“| x==”def”的字符变量。




Answer 1

匿名用户

下面是一个基于rlang的可能解决方案。

解释

>

首先，当时，我们需要使用列表all_条件为case_创建一个包含整个代码的字符串，这就是我的imap所做的。

其次，使用rlang::parse_quo，我们将字符串转换为Mutate中使用的表达式。

评论

listall_conditions元素的名称必须用反括号括起来。

library(tidyverse)
library(rlang)

df <- data.frame(x = 1:10)

all_conditions <- list(`1` = "x < 2",
                       `2` = "x >= 2 & x < 5",
                       `3` = "x >= 5 & x < 9",
                       `4` = "TRUE")

code <- imap(all_conditions, ~ str_c(.x, " ~ ", .y)) %>% 
          str_c(collapse = ", ") %>% str_c("case_when(",.,")")

df %>% 
  mutate(x2 = !!parse_quo(code, env = caller_env()))

#>     x x2
#> 1   1  1
#> 2   2  2
#> 3   3  2
#> 4   4  2
#> 5   5  3
#> 6   6  3
#> 7   7  3
#> 8   8  3
#> 9   9  4
#> 10 10  4