如何使用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)