如何使用R中的ifelse條件從向量中刪除

Question

我有一個值為 (1,2,3,4) 的向量 a 和另一個值為 (1,1,0,1) 的向量 b。 使用 b 中的元素作為標志，我想在元素 b 中找到 0 的相同位置從 A 中刪除向量元素。

  a <- c(1,2,3,4)
  b <- c(1,1,0,1)
     for(i in 1:length(b))
  {
    if(b[i] == 0)
    {
      a <- a[-i]
    }
  }

我得到了想要的輸出

[1] 1 2 4

但是使用 ifelse，我沒有得到所需的輸出。

    a <- c(1,2,3,4)
  b <- c(1,1,0,1)
    for(i in 1:length(b))
    {
      a <- ifelse(b[i] == 0,a[-i],a)
    }

輸出：

[1] 1

在這種情況下如何使用 ifelse？

Answer 1

我認為ifelse在這里不是正確的函數，因為ifelse給出與輸入相同長度的輸出，我們想在這里對值進行子集化。 您也不需要循環。 你可以直接做

a[b != 0]
#[1] 1 2 4

數據

a <- 1:4
b <- c(1, 1, 0, 1)

Answer 2

另一種選擇可能是：

a[as.logical(b)]

[1] 1 2 4

Answer 3

如果要使用ifelse ，可以使用以下代碼

na.omit(ifelse(b==0,NA,a))

以至於

> na.omit(ifelse(b==0,NA,a))
[1] 1 2 4
attr(,"na.action")
[1] 3
attr(,"class")
[1] "omit"

Answer 4

我們也可以使用雙重否定

a[!!b]
#[1] 1 2 4

數據

a <- 1:4  
b <- c(1, 1, 0, 1)