Как создать новую матрицу с условием одной переменной в R?

Question

У меня есть переменная x длины 20 как

x
 [1] 3 3 1 2 3 1 2 1 3 1 3 1 3 2 1 1 3 1 3 2.

Теперь я хотел бы создать матрицу порядка 20 by 2, где первый столбец W_i1 заполнен 1 если x==1 и -1, если x not equal to 1. Аналогично, второй столбец заполняется 1, если x=2, и -1, если x not equal to 2. Если x=3, оба столбца заполняются на -1. Вот ожидаемый результат

W_ij
      [,1] [,2]
 [1,]   -1   -1
 [2,]   -1   -1
 [3,]    1   -1
 [4,]   -1    1
 [5,]   -1   -1
 [6,]    1   -1
 [7,]   -1    1
 [8,]    1   -1
 [9,]   -1   -1
[10,]    1   -1
[11,]   -1   -1
[12,]    1   -1
[13,]   -1   -1
[14,]   -1    1
[15,]    1   -1
[16,]    1   -1
[17,]   -1   -1
[18,]    1   -1
[19,]   -1   -1
[20,]   -1    1

Я могу сделать это для каждого столбца отдельно и записать в виде матрицы

W_i1=ifelse(x==1,1,-1)
W_i2=ifelse(x==2,1,-1)
W_ij=matrix(c(W_i1,W_i2),
             nrow = 20,
             ncol =2,
             byrow = F)

Могу ли я сделать это, используя любой l oop?

Любая помощь приветствуется?

Answer 1

Самый простой способ преобразовать это

cbind(x == 1, x == 2) * 2 - 1

     [,1] [,2]
 [1,]   -1   -1
 [2,]   -1   -1
 [3,]    1   -1
 [4,]   -1    1
 [5,]   -1   -1
 [6,]    1   -1
 [7,]   -1    1
 [8,]    1   -1
 [9,]   -1   -1
[10,]    1   -1
[11,]   -1   -1
[12,]    1   -1
[13,]   -1   -1
[14,]   -1    1
[15,]    1   -1
[16,]    1   -1
[17,]   -1   -1
[18,]    1   -1
[19,]   -1   -1
[20,]   -1    1

Answer 2

Мы можем сделать за один шаг с

replace(cbind( c(-1, 1)[(x==1) + 1], c(-1, 1)[(x==2) + 1] ), x == 3, -1)
#      [,1] [,2]
# [1,]   -1   -1
# [2,]   -1   -1
# [3,]    1   -1
# [4,]   -1    1
# [5,]   -1   -1
# [6,]    1   -1
# [7,]   -1    1
# [8,]    1   -1
# [9,]   -1   -1
#[10,]    1   -1
#[11,]   -1   -1
#[12,]    1   -1
#[13,]   -1   -1
#[14,]   -1    1
#[15,]    1   -1
#[16,]    1   -1
#[17,]   -1   -1
#[18,]    1   -1
#[19,]   -1   -1
#[20,]   -1    1

---

Поскольку некоторые шаги похожи, можно создать функцию

f1 <- function(vec, v) c(-1, 1)[(vec == v) + 1]
replace(cbind(f1(x, 1), f1(x, 2)), x == 3, -1)

data

x <- c(3, 3, 1, 2, 3, 1, 2, 1, 3, 1, 3, 1, 3, 2, 1, 1, 3, 1, 3, 2)