Существует ли троичный оператор в R?

175

Поскольку вопрос задает, есть ли последовательность управления в R, подобная тернарному оператору C ? Если да, то как ты это используешь? Спасибо!

eykanal
источник
1
Вы хотите что-то более мощное, чем ifelse, или просто более компактную форму?
Карл Виттофт
@CarlWitthoft В основном более компактная форма; просто способ сохранить запись if (x>1) y=2 else y=3. Письмо y=однажды имеет определенную привлекательность для этого.
эйканал

Ответы:

302

Как ifэто функция Rи возвращает последнее вычисление, если-иначе эквивалентно ?:.

> 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

Шучу, вы можете определить с-стиль ?::

`?` <- 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.

kohske
источник
23

Как и все остальные, используйте 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"

(Это работает, потому что приоритет :ниже, чем ?.)

К сожалению, это нарушает существующие операторы справки и последовательности.

Ричи Коттон
источник
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(`?`)
Томми
источник
5

Я бы посмотрел на 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
Привет, Пол - ты хотел показать что-то на ifelseсвоем примере? ;)
Джош О'Брайен
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`
Andrie
источник
1
Как сказал @kohske, это тоже будет работать:print(if (x<2) "Less than" else "Greater than")
Бен Болкер
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, yи z. Первый вернет либо yили z(весь объект), а второй вернет соответствующий элемент y( condition2==TRUE) или z( condition2==FALSE).

Также обратите внимание , что ifelseбудет происходить медленнее , чем if/ elseесли condition, yи zвсе векторы с длиной 1.

Джошуа Ульрих
источник
спасибо Джошуа, ваш ответ очень помог, я нашел ответ из поста, который вы упомянули stackoverflow.com/a/8792474/3019570
Махди Джадалиха
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
UpsideDownRide
источник