Question

我希望能够以编程方式使用dplyr case_when来替换基本R cut()函数。

目前，case_when可以通过NSE与外部参数一起使用，如：

library(dplyr)
library(rlang)

patterns <- list(
  x <= 2 ~ "<=2",
  x <= 4 ~ "2<->4",
  x > 4 ~ ">4"
 )
 x <- 1:10
 case_when(!!!patterns)

我想要做的是：在变异

中使用另一个变量

这个想法会是这样的，虽然我无法弄清楚如何让它发挥作用：

library(dplyr)
patterns_lazy <- list(
  !!quo(x) <= 2 ~ "<=2",
  !!quo(x) <= 4 ~ "2<->4",
  !!quo(x) > 4 ~ ">4"
)
x <- "cyl"
mtcars %>% mutate(ABC = case_when(!!!patterns_lazy))

我希望能够定义我想要过滤的列（在字符串内），并检索这样的内容（此示例不能正常工作，因为它是所需的语法）：

x <- "cyl"
mtcars %>%
  select(cyl) %>%
  mutate(ABC = case_when(!!!patterns_lazy)) %>%
  head()

  cyl ABC
1   6 >4
2   6 >4
3   4 2<->4
4   6 >4
5   8 >4
6   6 >4

感谢您的帮助：）

Answer 1

你不能在那里使用!!：

patterns <- list(
  !!quo(x) <= 2 ~ "<=2",
  !!quo(x) <= 4 ~ "2<->4",
  !!quo(x) > 4 ~ ">4"
)

list()和~都不支持quasiquotation。
如果确实支持quasiquotation，则需要注意运算符优先级并将!!quo()括在括号中。
最后，对x的引用将评估为一个字符串，你要将数字与字符串进行比较（在你的例子中它将是"cyl），由于隐式强制，R将会愉快地做：/

因此您需要使用exprs()代替list()，并使用x .data代词而不是引用x。

exprs()将创建一个未评估的表达式列表。未评估是好的：如果你的公式被评估，它将带有一个环境（这里是全局环境）并且该环境不包含任何内容提供给dplyr的数据，特别是没有.data 代词。另一方面，如果公式是“无上下文”，则它们在我们想要的数据上下文中进行评估。

patterns_lazy <- exprs(
  .data[[x]] <= 2 ~ "<=2",
  .data[[x]] <= 4 ~ "2<->4",
  .data[[x]] > 4 ~ ">4"
)

x <- "cyl"
pull(mutate(mtcars, case_when(!!!patterns_lazy)))
#>  [1] ">4"    ">4"    "2<->4" ">4"    ">4"    ">4"    ">4"    "2<->4" "2<->4"
#> [10] ">4"    ">4"    ">4"    ">4"    ">4"    ">4"    ">4"    ">4"    "2<->4"
#> [19] "2<->4" "2<->4" "2<->4" ">4"    ">4"    ">4"    ">4"    "2<->4" "2<->4"
#> [28] "2<->4" ">4"    ">4"    ">4"    "2<->4"

Answer 2

以下是ifelse

的一个选项

f1 <- function(data, x){
       x <- enquo(x)       
       f2 <- function(y) ifelse(y <= 2, "<=2", ifelse(y <=4, "2<->4", ">4"))

  data %>%
          mutate( ABC = f2(UQ(x)))
 }

f1(mtcars, cyl) %>%
              head()
#   mpg cyl disp  hp drat    wt  qsec vs am gear carb   ABC
#1 21.0   6  160 110 3.90 2.620 16.46  0  1    4    4    >4
#2 21.0   6  160 110 3.90 2.875 17.02  0  1    4    4    >4
#3 22.8   4  108  93 3.85 2.320 18.61  1  1    4    1 2<->4
#4 21.4   6  258 110 3.08 3.215 19.44  1  0    3    1    >4
#5 18.7   8  360 175 3.15 3.440 17.02  0  0    3    2    >4
#6 18.1   6  225 105 2.76 3.460 20.22  1  0    3    1    >4

以编程方式使用dplyr :: case_when参数

2 个答案: