ifelse在向量的每个元素上

问题描述 投票:0回答:1

看看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创建于2019-03-13(v0.2.1)

任何帮助将不胜感激。

r if-statement
1个回答
6
投票

ifelse不是一个向量的函数,它是3个相同长度的向量的函数。第一个矢量,称为test,是一个布尔值,第二个矢量yes和第三个矢量no给出结果中的元素,根据test值逐项选择。

size = 1的样本与test的大小不同(除非test的长度为1),因此它将由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)
)

矢量实际上不必具有相同的长度。帮助页面?ifelse解释说:“如果yesno太短,他们的元素将被回收。”这是您观察到的行为“它为每种情况生成一次,并始终返回非常随机的数字。”。

© www.soinside.com 2019 - 2024. All rights reserved.