向量的每个元素上的ifelse

时间:2019-03-13 17:53:45

标签: r if-statement

看着this post,我认为ifelsef(c(x1, x2, x3)) = c(f(x1), f(x2), f(x3))的意义上是矢量化的。

因此,我认为z1的代码(下面提供)是否将对向量y的每个元素执行以下操作:

  • 测试是否统一。
    • 如果,请从{1、3、5、7、9}中抽取一个随机数。
    • 如果,请从{0,2,4,6,8}中抽取一个随机数。

但是,不幸的是它没有那样做。它为每种情况生成一次,并始终返回该非常随机的数字。

我到底在哪里做错了?或者,实际上是ifelse的预期行为吗?

请注意,如果我将其用作sapply中的包装函数,则会得到预期的输出z2(从某种意义上说,它不是确定性的z1每种情况的发生就足够了),如下所示。

y <- rbinom(n = 20,
            size = 1,
            prob = 0.5)

z1 <- ifelse(test = (y == 1),
             yes = sample(x = c(1, 3, 5, 7, 9),
                          size = 1),
             no = sample(x = c(0, 2, 4, 6, 8),
                         size = 1))

z2 <- sapply(X = y,
             FUN = function(w)
             {
               ifelse(test = (w == 1),
                      yes = sample(x = c(1, 3, 5, 7, 9),
                                   size = 1),
                      no = sample(x = c(0, 2, 4, 6, 8),
                                  size = 1))
             })

data.frame(y, z1, z2)
#>    y z1 z2
#> 1  0  2  2
#> 2  1  1  3
#> 3  1  1  9
#> 4  1  1  7
#> 5  0  2  0
#> 6  0  2  2
#> 7  1  1  7
#> 8  1  1  7
#> 9  0  2  0
#> 10 1  1  5
#> 11 0  2  0
#> 12 0  2  0
#> 13 0  2  6
#> 14 0  2  0
#> 15 0  2  2
#> 16 1  1  7
#> 17 1  1  7
#> 18 0  2  2
#> 19 0  2  2
#> 20 0  2  0

unique(x = z1[y == 1])
#> [1] 1

unique(x = z1[y == 0])
#> [1] 2

reprex package(v0.2.1)于2019-03-13创建

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:6)

ifelse不是一个向量的函数,而是 3个相同长度的向量的函数。第一个向量称为test,是布尔值,第二个向量yes和第三个向量no给出结果中的元素,并根据{{1 }}值。

test的样本大小与size = 1不同(除非test的长度为1),因此test将对其进行回收(请参见下面的注释) )。相反,请从一开始就绘制与ifelse相同大小的样本:

test

向量实际上不必具有相同的长度。帮助页面ifelse( test = (y == 1), yes = sample(x = c(1, 3, 5, 7, 9), size = length(y), replace = TRUE), no = sample(x = c(0, 2, 4, 6, 8), size = lenght(y), replace = TRUE) ) 解释:“如果?ifelseyes太短,则它们的元素将被回收。”这是您在中观察到的行为“它为每种情况生成一次,并始终返回该非常随机的数字。”。