Как создать фиктивные переменные для группы другой переменной в Tidyverse - PullRequest
1 голос
/ 27 июня 2019

Я хочу создать (фиктивные) переменные, которые показывают, входит ли наблюдение в группу наблюдений (идентифицируемую по общему идентификатору группы) с определенной комбинацией характеристик в этой группе.Пример кода проясняет, что именно я имею в виду.

Я пробовал комбинации group_by и caret :: dummyVars, но безуспешно.У меня заканчиваются идеи - любая помощь будет принята с благодарностью.

library(tidyverse)

# Input data
# please note: in my case each value of the column Role will appear only once per Group_ID.

input_data <- tribble( ~Group_ID, ~Role, ~Income,
                        #--|--|----
                        1, "a", 3.6,
                        1, "b", 8.5,

                        2, "a", 7.6,
                        2, "c", 9.5,
                        2, "d", 9.7,

                        3, "a", 1.6,
                        3, "b", 4.5,
                        3, "c", 2.7,
                        3, "e", 7.7,

                        4, "b", 3.3,
                        4, "c", 6.2,
)

# desired output
output_data <- tribble( ~Group_ID, ~Role, ~Income, ~Role_A,  ~Role_B, ~Role_C, ~Role_D, ~Role_E, ~All_roles,
                        #--|--|----
                        1, "a", 3.6, 1, 1, 0, 0, 0, "ab",
                        1, "b", 8.5, 1, 1, 0, 0, 0, "ab",

                        2, "a", 7.6, 1, 0, 1, 1, 0, "acd",
                        2, "c", 9.5, 1, 0, 1, 1, 0, "acd",
                        2, "d", 9.7, 1, 0, 1, 1, 0, "acd",

                        3, "a", 1.6, 1, 1, 1, 0, 1, "abce",
                        3, "b", 4.5, 1, 1, 1, 0, 1, "abce",
                        3, "c", 2.7, 1, 1, 1, 0, 1, "abce",
                        3, "e", 7.7, 1, 1, 1, 0, 1, "abce",

                        4, "b", 3.3, 0, 1, 1, 0, 0, "bc",
                        4, "c", 6.2, 0, 1, 1, 0, 0, "bc"
)

Ответы [ 2 ]

1 голос
/ 27 июня 2019

Далее используются базовые функции моделирования R для создания макетов.

Сначала создайте матрицу модели без перехвата.

fit <- lm(Group_ID ~ 0 + Role, input_data)
m <- model.matrix(fit)

Теперь обработайте эту матрицу, отметив, что пустышки, о которых идет речь в вопросе, являются суммами по группам Group_ID.

input_data %>%
  bind_cols(m %>% as.data.frame()) %>%
  group_by(Group_ID) %>%
  mutate_at(vars(matches("Role[[:alpha:]]")), sum) %>%
  mutate(all_roles = paste(Role, collapse = ""))
## A tibble: 11 x 9
## Groups:   Group_ID [4]
#   Group_ID Role  Income Rolea Roleb Rolec Roled Rolee all_roles
#      <dbl> <chr>  <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <chr>    
# 1        1 a        3.6     1     1     0     0     0 ab       
# 2        1 b        8.5     1     1     0     0     0 ab       
# 3        2 a        7.6     1     0     1     1     0 acd      
# 4        2 c        9.5     1     0     1     1     0 acd      
# 5        2 d        9.7     1     0     1     1     0 acd      
# 6        3 a        1.6     1     1     1     0     1 abce     
# 7        3 b        4.5     1     1     1     0     1 abce     
# 8        3 c        2.7     1     1     1     0     1 abce     
# 9        3 e        7.7     1     1     1     0     1 abce     
#10        4 b        3.3     0     1     1     0     0 bc       
#11        4 c        6.2     0     1     1     0     0 bc      
1 голос
/ 27 июня 2019

Использование dplyr и cSplit_e из splitstackshape.Для каждого Group_ID мы вставляем Role вместе, а затем разделяем их на новые столбцы двоичного значения на основе их присутствия и отсутствия, используя cSplit_e.

library(splitstackshape)
library(dplyr)

input_data %>%
    group_by(Group_ID) %>%
    mutate(new_role = paste(Role, collapse = "")) %>%
    ungroup() %>%
    cSplit_e("new_role", sep = "", type = "character", fill = 0)

#   Group_ID Role Income new_role new_role_a new_role_b new_role_c new_role_d new_role_e
#1         1    a    3.6       ab          1          1          0          0          0
#2         1    b    8.5       ab          1          1          0          0          0
#3         2    a    7.6      acd          1          0          1          1          0
#4         2    c    9.5      acd          1          0          1          1          0
#5         2    d    9.7      acd          1          0          1          1          0
#6         3    a    1.6     abce          1          1          1          0          1
#7         3    b    4.5     abce          1          1          1          0          1
#8         3    c    2.7     abce          1          1          1          0          1
#9         3    e    7.7     abce          1          1          1          0          1
#10        4    b    3.3       bc          0          1          1          0          0
#11        4    c    6.2       bc          0          1          1          0          0
...