삼항 연산자가 R에 있습니까?


175

질문에서 알 수 있듯이 R에 C의 삼항 연산자 와 비슷한 제어 시퀀스가 있습니까? 그렇다면 어떻게 사용합니까? 감사!


1
보다 강력한 무언가를 원하십니까 ifelse, 아니면 더 단순한 형태를 원하십니까?
Carl Witthoft

@CarlWitthoft 대부분 더 컴팩트 한 형태; 단순히 글쓰기를 절약하는 방법 if (x>1) y=2 else y=3. y=한 번 쓰는 것은 그것에 호소합니다.
eykanal

답변:


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

농담으로 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 언어의 내부 학습에 도움이 될 수 있습니다.


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당신의 모범으로 뭔가를 보여 주려고 했습니까? ;)
Josh O'Brien

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`

1
@kohske 말했듯이,이 역시 작동합니다 :print(if (x<2) "Less than" else "Greater than")
벤 Bolker

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 인 모든 벡터입니다.


감사합니다 Joshua, 귀하의 답변이 많이 도움이
되었으므로,

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
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.