Использовать значения матриц в качестве индексов

Question

Я выполняю проверку перестановки Фишера, в которой я должен генерировать все комбинации статуса лечения.

У нас есть 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

Публикация мой комментарий в качестве решения. Это модификация предложения Хероки. + преобразует 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 

Answer 2

Использование базового 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 3

Как насчет:

out <- matrix(0L, nrow = nrow(x), ncol = max(x)) 

for (i in 1:nrow(x)) out[i, x[i, ]] <- 1L 

> out 
    [,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 

Мы также можем сделать это без цикла, хотя это может быть немного менее читаемым (спасибо @Frank для ног):

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

Легко упаковывается в функции:

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