在dplyr的mutate中使用switch语句 [英] Using switch statement within dplyr's mutate

问题描述

我想在dplyr的mutate中使用switch语句.我有一个简单的函数，可以执行一些操作并通过switch分配备用值，例如:

I would like to use a switch statement within dplyr's mutate. I have a simple function that performs some operations and assigns alternative values via switch, for example:

convert_am <- function(x) {
    x <- as.character(x)
    switch(x,
           "0" = FALSE,
           "1" = TRUE,
           NA)
}

这在应用于标量时可以按需工作:

This works as desired when applied to scalars:

>> convert_am(1)
[1] TRUE
>> convert_am(2)
[1] NA
>> convert_am(0)
[1] FALSE

我想通过 mutate 调用获得相同的结果:

I would like to arrive at equivalent results via mutate call:

mtcars %>% mutate(am = convert_am(am))

此操作失败:

mutate_impl(.data，点)中的错误:评估错误:EXPR必须是长度为1的向量.

Error inmutate_impl(.data, dots) : Evaluation error: EXPR must be a length 1 vector.

我知道这是因为传递给switch ar的值不是单一的，例如:

I understand that this is because values passed to switch ar not single, as in example:

convert_am(c(1,2,2)) switch(x，0 = FALSE，1 = TRUE，NA)中的错误:EXPR必须为长度1个向量

convert_am(c(1,2,2)) Error in switch(x, 0 = FALSE, 1 = TRUE, NA) : EXPR must be a length 1 vector

向量化

尝试向量化也会产生预期的结果:

Vectorization

Attempt to vectorize also yield the desired results:

convert_am <- function(x) {
    x <- as.character(x)

    fun_switch <- function(x) {
        switch(x,
               "0" = FALSE,
               "1" = TRUE,
               NA)
    }

    vf <- Vectorize(fun_switch, "x")
}

>> mtcars %>% mutate(am = convert_am(am))
Error in mutate_impl(.data, dots) : 
  Column `am` is of unsupported type function

注释

• 我知道dplyr中的 case_when ，并且我对使用它不感兴趣，我只想让 switch 在mutate中工作
• 理想的解决方案将允许进一​​步扩展以将 mutate_at 与以 .
传递的变量一起使用

Notes

• I'm aware of case_when in dplyr and I'm not interested in using it, I'm only interested in making switch work inside mutate
• Ideal solution would allow for further expansion to use mutate_at with variables passed as .

推荐答案

switch 未向量化，因此为了提高效率，您需要使用 ifelse 或 case_when -但是由于您的问题专门针对 switch ，因此您可以通过矢量化来实现所需的功能，例如

switch is not vectorized so for efficiency you need to use ifelse or case_when - but as your question is specifically about switch, you can achieve what you want by vectorizing, e.g.

convert_am <- Vectorize(function(x) {
    x <- as.character(x)
    switch(x,
       "0" = FALSE,
       "1" = TRUE,
       NA)
})

或

convert_am <- function(x) {
    x <- as.character(x)
    sapply(x, function(xx) switch(xx,
       "0" = FALSE,
       "1" = TRUE,
       NA))
}

它们都效率低下，因为它们涉及引擎盖下的循环.

They are both inefficient as they involve a loop under the hood.

这篇关于在dplyr的mutate中使用switch语句的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！