如何在R函数中使用switch语句?

时间:2012-05-01 04:19:58

标签: r switch-statement

我想在语句switch()中使用我的函数来根据函数参数的值触发不同的计算。

例如,在Matlab中,您可以通过编写

来实现
switch(AA)        
case '1'   
...   
case '2'   
...   
case '3'  
...  
end

我发现这篇帖子 - switch() statement usage - 解释了如何使用switch,但对我没有帮助,因为我想执行更复杂的计算(矩阵运算)而不是简单{{1} }}

4 个答案:

答案 0 :(得分:87)

嗯,switch可能不是真的意味着这样做,但你可以:

AA = 'foo'
switch(AA, 
foo={
  # case 'foo' here...
  print('foo')
},
bar={
  # case 'bar' here...
  print('bar')    
},
{
   print('default')
}
)

...每个案例都是一个表达式 - 通常只是一个简单的事情,但在这里我使用一个卷曲块,这样你就可以填写你想要的任何代码...

答案 1 :(得分:38)

我希望这个例子有所帮助。您可以使用花括号来确保切换器更换器中包含所有内容(抱歉不知道技术术语,但在=符号之前的术语会改变发生的情况)。我认为switch是一组受控制的if () {} else {}语句。

每次切换功能相同但我们提供的命令都会改变。

do.this <- "T1"

switch(do.this,
    T1={X <- t(mtcars)
        colSums(mtcars)%*%X
    },
    T2={X <- colMeans(mtcars)
        outer(X, X)
    },
    stop("Enter something that switches me!")
)
#########################################################
do.this <- "T2"

switch(do.this,
    T1={X <- t(mtcars)
        colSums(mtcars)%*%X
    },
    T2={X <- colMeans(mtcars)
        outer(X, X)
    },
    stop("Enter something that switches me!")
)
########################################################
do.this <- "T3"

switch(do.this,
    T1={X <- t(mtcars)
        colSums(mtcars)%*%X
    },
    T2={X <- colMeans(mtcars)
        outer(X, X)
    },
    stop("Enter something that switches me!")
)

这里面是一个函数:

FUN <- function(df, do.this){
    switch(do.this,
        T1={X <- t(df)
            P <- colSums(df)%*%X
        },
        T2={X <- colMeans(df)
            P <- outer(X, X)
        },
        stop("Enter something that switches me!")
    )
    return(P)
}

FUN(mtcars, "T1")
FUN(mtcars, "T2")
FUN(mtcars, "T3")

答案 2 :(得分:36)

切换的各种方式 ......

# by index
switch(1, "one", "two")
## [1] "one"


# by index with complex expressions
switch(2, {"one"}, {"two"})
## [1] "two"


# by index with complex named expression
switch(1, foo={"one"}, bar={"two"})
## [1] "one"


# by name with complex named expression
switch("bar", foo={"one"}, bar={"two"})
## [1] "two"

答案 3 :(得分:1)

这是对R中缺少的“选择cond1,stmt1,... else stmtelse”构造的更一般的答案。虽然有点笨拙,但其工作方式类似于C中存在的switch语句。

while (TRUE) { if (is.na(val)) { val <- "NULL" break } if (inherits(val, "POSIXct") || inherits(val, "POSIXt")) { val <- paste0("#", format(val, "%Y-%m-%d %H:%M:%S"), "#") break } if (inherits(val, "Date")) { val <- paste0("#", format(val, "%Y-%m-%d"), "#") break } if (is.numeric(val)) break val <- paste0("'", gsub("'", "''", val), "'") break }