Генерация переменных взаимодействия в кадрах данных R

Существует ли путь - кроме для цикла - для генерации новых переменных в кадре данных R, который будет всеми возможными взаимодействиями с 2 путями между существующими? т.е. предполагая кадр данных с тремя числовыми переменными V1, V2, V3, я хотел бы генерировать следующие новые переменные:

Inter.V1V2 (= V1 * V2) 
Inter.V1V3 (= V1 * V3)
Inter.V2V3 (= V2 * V3)

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

x <- read.table(textConnection('
   V1 V2 V3 V4
1  9   25   18
2  5   20   10
3  4   30   12
4  4   34   16'
), header=TRUE)

dim.init <- dim(x)[2]
for (i in 1: (dim.init - 1) ) {
        for (j in (i + 1) : (dim.init) ) {
                x[dim(x)[2] + 1]    <- x[i] * x[j]
                names(x)[dim(x)[2]] <- paste("Inter.V",i,"V",j,sep="")

        }
}
22
задан George Dontas 17 January 2010 в 13:07
поделиться

2 ответа

Вот один лайнер для вас, который также работает, если у вас есть факторы:

> model.matrix(~(V1+V2+V3+V4)^2,x)
  (Intercept) V1 V2 V3 V4 V1:V2 V1:V3 V1:V4 V2:V3 V2:V4 V3:V4
1           1  1  9 25 18     9    25    18   225   162   450
2           1  2  5 20 10    10    40    20   100    50   200
3           1  3  4 30 12    12    90    36   120    48   360
4           1  4  4 34 16    16   136    64   136    64   544
attr(,"assign")
 [1]  0  1  2  3  4  5  6  7  8  9 10
35
ответ дан 29 November 2019 в 04:06
поделиться

Вот так, используя combn и применим:

> x2 <- t(apply(x, 1, combn, 2, prod))

Установка имен столбцов может быть выполнена с помощью двух команд вставить :

> colnames(x2) <- paste("Inter.V", combn(1:4, 2, paste, collapse="V"), sep="")

Наконец, если вы хотите, чтобы все ваши переменные были вместе, просто cbind их:

> x <- cbind(x, x2)
>   V1 V2 V3 V4 Inter.V1V2 Inter.V1V3 Inter.V1V4 Inter.V2V3 Inter.V2V4 Inter.V3V4
1  1  9 25 18          9         25         18        225        162        450
2  2  5 20 10         10         40         20        100         50        200
3  3  4 30 12         12         90         36        120         48        360
4  4  4 34 16         16        136         64        136         64        544
15
ответ дан 29 November 2019 в 04:06
поделиться
Другие вопросы по тегам:

Похожие вопросы: