Question

我正在做Fisher的排列测试，我必须在其中生成治疗状态的所有组合。

我们有4个科目，其中2个接受治疗。使用combn，我可以生成所有受治疗科目的组合。例如，第一行表示处理第一和第二主题。

t(combn(4, 2))

     [,1] [,2]
[1,]    1    2
[2,]    1    3
[3,]    1    4
[4,]    2    3
[5,]    2    4
[6,]    3    4

如何从此矩阵转到治疗状态矩阵，如下所示：

      [,1] [,2] [,3] [,4]
[1,]    1    1    0   0
[2,]    1    0    1   0
...

Answer 1

使用base-R：

res <- t(apply(t(combn(4,2)),MARGIN=1,FUN=function(x){
  return(as.numeric(1:4 %in% x))
}))
> res
     [,1] [,2] [,3] [,4]
[1,]    1    1    0    0
[2,]    1    0    1    0
[3,]    1    0    0    1
[4,]    0    1    1    0
[5,]    0    1    0    1
[6,]    0    0    1    1

Answer 2

如何：

m <- matrix(0L, choose(4, 2), 4)
m[cbind(rep(1:choose(4, 2), each = 2), c(combn(4, 2)))] <- 1L

我们也可以在没有循环的情况下做到这一点，虽然可能性稍差（感谢@Frank的步法）：

participants <- function(m, n){
  if (n > m) stop( )
  mcn <- choose(m, n)
  out <- matrix(0L, mcn, m)
  out[cbind(rep(1:mcn, each = n), c(combn(m, n)))] <- 1L
  out
}

> participants(6, 5)
     [,1] [,2] [,3] [,4] [,5] [,6]
[1,]    1    1    1    1    1    0
[2,]    1    1    1    1    0    1
[3,]    1    1    1    0    1    1
[4,]    1    1    0    1    1    1
[5,]    1    0    1    1    1    1
[6,]    0    1    1    1    1    1

轻松融入功能：

#include <stdlib.h>

Answer 3

将我的评论作为解决方案发布。这是对@Heroka的建议的修改。 +会将logical转换为numeric，并且应该比as.integer更快。

+(t(combn(4,2, FUN=function(x) 1:4 %in% x)))
#     [,1] [,2] [,3] [,4]
#[1,]    1    1    0    0
#[2,]    1    0    1    0
#[3,]    1    0    0    1
#[4,]    0    1    1    0
#[5,]    0    1    0    1
#[6,]    0    0    1    1

使用矩阵值作为索引

3 个答案: