循环中的随机数

时间:2012-05-12 20:28:21

标签: r random for-loop

这让我疯狂,到现在为止,我绝对没有希望:(

这是矩阵draw(不要注意NAs,后来我需要这些列):

      [,1] [,2] [,3] [,4] [,5]  
[1,]    1    0   NA   NA   NA  
[2,]    2   20   NA   NA   NA  
[3,]    2   30   NA   NA   NA  
[4,]    2   40   NA   NA   NA  
[5,]    1   50   NA   NA   NA  
[6,]    2   70   NA   NA   NA  
[7,]    2   80   NA   NA   NA  
[8,]    2   90   NA   NA   NA  
[9,]    1  100   NA   NA   NA
[10,]   2  120   NA   NA   NA
[11,]   2  130   NA   NA   NA
[12,]   2  140   NA   NA   NA

所以,根据第1列,我需要在正常分布的第3列中写入数字。但在6%的情况下,它必须为0。

这就是我想要解决的问题(现在只有在draw[i,1]==2时才会这样做):

for (i in 1:nrow(draw))
{
    p <- runif(1, min=0, max=1)
    if (draw[i,2]==2 && p>0.06) 
    {   
        draw[i,3] <- rnorm(1, mean=17, sd=7.5)} else {draw[i,3]<-0}

    if (draw[i,1]==1) {draw[i,3] <- rnorm(1, mean=11, sd=3)}
    if (draw[i,1]==3) {draw[i,3] <- rnorm(1, mean=17, sd=15)}
    if (draw[i,3]<1 && draw[i,3]!=0) {draw[i,3] <- 1} else {draw[i,3] <- ceiling(draw[i,3])
}
}

并出现问题:  我注意到p(我猜这必须是随机数)总是一样的  2.所有绘制[i,3]其中draw [1,2] == 2都是0.这很奇怪,因为它甚至在p> 0.06时发生:

      [,1] [,2] [,3] [,4] [,5]
 [1,]    1    0   19   NA   NA
 [2,]    2   20    0   NA   NA
 [3,]    2   30    0   NA   NA
 [4,]    2   40    0   NA   NA
 [5,]    1   50   11   NA   NA
 [6,]    2   70    0   NA   NA
 [7,]    2   80    0   NA   NA
 [8,]    2   90    0   NA   NA
 [9,]    1  100    7   NA   NA
[10,]    2  120    0   NA   NA
[11,]    2  130    0   NA   NA
[12,]    2  140    0   NA   NA

为什么?怎么了?

1 个答案:

答案 0 :(得分:3)

嗯。你说draw[i,1]==2当你想要做某事时,但在你的代码中你写draw[i,2]==2 ......这可能是问题所在!

# This line might work better:
if (draw[i,1]==2 && p>0.06) 

......但无论如何它似乎相当混乱。也许这就是你的意思? - 禁止我放在那里的任何错误:)

for (i in 1:nrow(draw)) {
  p <- runif(1, min=0, max=1)

  draw[i,3] <- if (p <= 0.06) 0
    else switch(draw[i,1], 
      rnorm(1, mean=11, sd=3),   # 1
      rnorm(1, mean=17, sd=7.5), # 2
      rnorm(1, mean=17, sd=15)   # 3
    )

  draw[i,3] <- if (draw[i,3]<1 && draw[i,3]!=0) 1 else ceiling(draw[i,3])
}