随机分配不同的值，以作为R

Question

使用不同的概率行具有这样的数据帧：

ID var 
1 NA 
2 NA 
3 NA 
4 NA 
... 

我需要随机地分配20个％的行var值是A，和30％的行为B，和50％的行是C.

有没有一些有效的方法来解决这个问题？

Answer 1

假设你有数据框名为DF： 那么你可以这样写：

randvar = sample(c('A','B','C'),size = nrow(df),prob = c(0.2,0.3,0.5),replace = TRUE) 
df$var = randvar 

假设你想要的 “A” s是正确的20％％，因此，做30％ “B” 和 “C”在50％ 那么它是不是一个行代码，假设你的C（0.2,0.3,0.5）* df_size是所有整我的回答是：

n = nrow(df) 
df$var = "C" #initialize all value to be "C" 
index = 1:n 
indexa = sample(index,0.2*n) #pick 20% index for "A" 
indexb = sample(index[-indexa],0.3*n) #pick 30% index for "B" need to rule out the "A"s you already picked 
df$var[indexa] = "A" #assign "A" to df$var at indexa 
df$var[indexb] = "B" #assign "B" to df$var at indexb 
#the rest 50% is "C"