如何按因子生成随机治疗变量？

Question

定义

x <- data.frame(
     ID=letters[1:10],
     class = as.factor(c(rep(1,5),rep(2,5))),
     treat = rep(0,10))

st

> x
   ID class treat
1   a     1     0
2   b     1     0
3   c     1     0
4   d     1     0
5   e     1     0
6   f     2     0
7   g     2     0
8   h     2     0
9   i     2     0
10  j     2     0

我有两个级别的治疗，1 和 2。 2. 我想为每个级别 st 分配每个班级一个单位，随机化后，我们得到如下内容：

> x
   ID class treat
1   a     1     0
2   b     1     0
3   c     1     1
4   d     1     0
5   e     1     2
6   f     2     0
7   g     2     0
8   h     2     0
9   i     2     2
10  j     2     1 

st 单位 c 和 j 获得 1 级治疗以及e和i级别2。

如何在R中生成治疗向量？

Answer 1

我假设您只想在每个班级中分配一个 1 级治疗和一个 2 级治疗。您可以使用 plyr 包中的 ddply 函数轻松完成此操作：

  set.seed(1)
  require(plyr)
> ddply(x, .(class), transform, 
        treat = replace(treat, sample(seq_along(treat),2), 1:2))

   ID class treat
1   a     1     0
2   b     1     1
3   c     1     0
4   d     1     0
5   e     1     2
6   f     2     0
7   g     2     0
8   h     2     1
9   i     2     2
10  j     2     0

解释一下：ddply 函数按 < code>class 变量，并且在每个数据帧内，它通过用 1 和 2 替换 2 个随机选择的条目来“转换”treat 列。代码>样本(...,2)函数在 treat 列中选取两个随机索引。其他变体（例如，为每种治疗类型分配 1 个以上）可以类似地完成。