R中是否存在三元运算符?

时间:2012-01-09 14:44:25

标签: r operators

正如问题所问,R中的控制序列是否与C ternary operator相似?如果是这样,你如何使用它?谢谢!

8 个答案:

答案 0 :(得分:263)

由于ifR中的函数并返回最新评估,if-else等同于?:

> a <- 1
> x <- if(a==1) 1 else 2
> x
[1] 1
> x <- if(a==2) 1 else 2
> x
[1] 2

R的力量是矢量化。三元运算符的向量化是ifelse

> a <- c(1, 2, 1)
> x <- ifelse(a==1, 1, 2)
> x
[1] 1 2 1
> x <- ifelse(a==2, 1, 2)
> x
[1] 2 1 2

开个玩笑,你可以定义c风格的?:

`?` <- function(x, y)
    eval(
      sapply(
        strsplit(
          deparse(substitute(y)), 
          ":"
      ), 
      function(e) parse(text = e)
    )[[2 - as.logical(x)]])

在这里,您不需要关注括号:

> 1 ? 2*3 : 4
[1] 6
> 0 ? 2*3 : 4
[1] 4
> TRUE ? x*2 : 0
[1] 2
> FALSE ? x*2 : 0
[1] 0

但你需要括号来分配:(

> y <- 1 ? 2*3 : 4
[1] 6
> y
[1] 1
> y <- (1 ? 2*3 : 4)
> y
[1] 6

最后,你可以用c:

做非常相似的方式
`?` <- function(x, y) {
  xs <- as.list(substitute(x))
  if (xs[[1]] == as.name("<-")) x <- eval(xs[[3]])
  r <- eval(sapply(strsplit(deparse(substitute(y)), ":"), function(e) parse(text = e))[[2 - as.logical(x)]])
  if (xs[[1]] == as.name("<-")) {
    xs[[3]] <- r
        eval.parent(as.call(xs))
  } else {
    r
  }
}       

你可以摆脱括号:

> y <- 1 ? 2*3 : 4
> y
[1] 6
> y <- 0 ? 2*3 : 4
> y
[1] 4
> 1 ? 2*3 : 4
[1] 6
> 0 ? 2*3 : 4
[1] 4

这些不是日常使用,但可能有助于学习R语言的一些内部。

答案 1 :(得分:19)

与其他人一样,使用ifelse,但您可以定义运算符,以便几乎拥有三元运算符语法。

`%?%` <- function(x, y) list(x = x, y = y)
`%:%` <- function(xy, z) if(xy$x) xy$y else z

TRUE %?% rnorm(5) %:% month.abb
## [1]  0.05363141 -0.42434567 -0.20000319  1.31049766 -0.31761248
FALSE %?% rnorm(5) %:% month.abb
## [1] "Jan" "Feb" "Mar" "Apr" "May" "Jun" "Jul" "Aug" "Sep" "Oct" "Nov" "Dec"
# or, more generally
condition %?% value1 %:% value2

如果你定义没有%符号的运算符,它实际上是有效的,所以你可以有

`?` <- function(x, y) if(x) y[[1]] else y[[2]]
`:` <- function(y, z) list(y, z)

TRUE ? rnorm(5) : month.abb
## [1]  1.4584104143  0.0007500051 -0.7629123322  0.2433415442  0.0052823403
FALSE ? rnorm(5) : month.abb
## [1] "Jan" "Feb" "Mar" "Apr" "May" "Jun" "Jul" "Aug" "Sep" "Oct" "Nov" "Dec"

(这是有效的,因为:的优先级低于?。)

不幸的是,这会破坏现有的帮助和序列运算符。

答案 2 :(得分:5)

就像一个恶作剧一样,你可以重新定义?运算符(几乎)像三元运算符一样工作(这是一个糟糕的想法):

`?` <- function(x, y) { y <-substitute(y); if(x) eval(y[[2]], parent.frame()) else eval(y[[3]], parent.frame()) }

x <- 1:3
length(x) ? (x*2) : 0
x <- numeric(0)
length(x) ? (x*2) : 0

for(i in 1:5) cat(i, (i %% 2) ? "Odd\n" : "Even\n")

...但是您需要将表达式放在括号中,因为默认优先级与C中的不同。

请记住在完成游戏后恢复旧的帮助功能:

rm(`?`)

答案 3 :(得分:4)

我会看一下ifelse命令。我会称之为更好,因为它也是矢量化的。使用汽车数据集的示例:

> cars$speed > 20
 [1] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
[13] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
[25] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
[37] FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE  TRUE  TRUE  TRUE  TRUE
[49]  TRUE  TRUE

> ifelse(cars$speed > 20, 'fast', 'slow')
 [1] "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow"
[11] "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow"
[21] "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow"
[31] "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow" "slow"
[41] "slow" "slow" "slow" "fast" "fast" "fast" "fast" "fast" "fast" "fast"

答案 4 :(得分:4)

您的链接指向if声明。

> x <- 1
> if(x < 2) print("Less than") else print("Greater than")
[1] "Less than"

如果您的输入变量是矢量,那么ifelse可能更合适:

> x <- 1:3
> ifelse(x<=2, "Less than or equal", "Greater than")
[1] "Less than or equal" "Less than or equal" "Greater than"   

要访问if的帮助页面,您需要在反引号中嵌入if

?`if`

ifelse的帮助页面位于:

`?ifelse`

答案 5 :(得分:4)

它没有明确存在,但你可以这样做:

set.seed(21)
y <- 1:10
z <- rnorm(10)

condition1 <- TRUE
x1 <- if(condition1) y else z

condition2 <- sample(c(TRUE,FALSE),10,TRUE)
x2 <- ifelse(condition2, y, z)

两者之间的区别在于condition1必须是长度为1的逻辑向量,而condition2必须是与x长度相同的逻辑向量,yz。第一个将返回yz(整个对象),而第二个将返回ycondition2==TRUE)或z的相应元素( condition2==FALSE)。

另请注意,如果ifelseifelse都是有长度的向量,则condition会慢于y / z 1。

答案 6 :(得分:2)

如果按照以下方式使用,

if的工作方式与非传播ifelse相同:

`if`(condition, doIfTrue, doIfFalse)

使用ifelse的优点是当矢量化阻碍时(即我有标量布尔值和列表/矢量事物)

ifelse(TRUE, c(1,2), c(3,4))
[1] 1
`if`(TRUE, c(1,2), c(3,4))
[1] 1 2

答案 7 :(得分:1)

我编写了一个小型语言扩展,它在 R 中模拟 C 的条件三元运算符。它可以作为包安装在 here

该实现基于 answer given by @kohske,但我进行了一些更改,以便在 if_trueif_false 参数包含冒号,允许链接条件语句并保留 ? 运算符的基本功能。

我会参考其他人关于重新定义运算符的危险的警告,但这是一个很好的例子,说明 R 语言的动态性有多强!