Создать набор переменных в фрейме данных на основе набора правил из другого фрейма данных

Question

Вопрос звучит немного обобщенно c, но я думаю, что пример будет намного понятнее:

У меня есть следующие два кадра данных

data1

group1     group2       group3     Level 
cat         cat          dog        1
dog         parrot       cat        1
mouse       dolphin      dolphin    1
red         blue         blue       2
green       yellow       green      2
black       purple       cat        2

data2

var1        level    Score
cat           1        1
dog           1        1
mouse         1        1
dolphin       1        0
parrot        1        1
red           2        1
blue          2        1
green         2        1
purple        2        1
cat           2        0
black         2        0
yellow        2        1

Я хочу изменить данные1, включив 3 новых столбца (по одному для каждой группы1, группы2 и группы3), со значениями, которые я найду в столбце «оценка» из данных2 в зависимости от уровня «уровень» (уровень является фактором). Поэтому в основном я хочу получить что-то вроде этого:

group1     group2       group3     Level      var1     var2     var3
cat         cat          dog        1          1        1        1
dog         parrot       cat        1          1        1        1
mouse       dolphin      dolphin    1          1        0        0
red         blue         blue       2          1        1        1
green       yellow       green      2          1        1        1
black       purple       cat        2          0        1        0

---

Пример данных

df1 <- structure(list(
  group1 = c("cat", "dog", "mouse", "red", "green", "black"),
  group2 = c("cat", "parrot", "dolphin", "blue", "yellow", "purple"),
  group3 = c("dog", "cat", "dolphin", "blue", "green", "cat"),
  Level = structure(c(1L, 1L, 1L, 2L, 2L, 2L), .Label = c("1", "2"), class = "factor")),
  row.names = c(NA, -6L), class = "data.frame")

df2 <- structure(list(
  var1 = c("cat", "dog", "mouse", "dolphin", "parrot", "red", "blue", "green", "purple", "cat", "black", "yellow"),
  level = structure(c(1L, 1L, 1L, 1L, 1L, 2L, 2L, 2L, 2L, 2L, 2L, 2L), .Label = c("1", "2"), class = "factor"),
  Score = c(1L, 1L, 1L, 0L, 1L, 1L, 1L, 1L, 1L, 0L, 0L, 1L)),
  row.names = c(NA, -12L), class = "data.frame")

Answer 1

Мы можем повернуть первый набор данных в «длинный» формат, объединить со вторым и затем повернуть его обратно в «широкий» формат

library(dplyr)
library(tidyr)
library(stringr)
df1 %>%
    mutate(rn = row_number()) %>%
    pivot_longer(cols  = -c(rn, Level), values_to = 'var1') %>% 
    rename(level = Level) %>% 
    left_join(df2) %>% 
    mutate(name = str_replace(name, 'group', 'varn')) %>% 
    na.omit %>%
    select(-level, -var1) %>% 
    pivot_wider(names_from = name, values_from = Score, values_fill = list(Score = 0)) %>% 
    select(-rn) %>% 
    bind_cols(df1, .)
#   group1  group2  group3 Level varn1 varn2 varn3
#1    cat     cat     dog     1     1     1     1
#2    dog  parrot     cat     1     1     0     1
#3  mouse dolphin dolphin     1     1     0     0
#4    red    blue    blue     2     1     1     1
#5  green  yellow   green     2     1     0     1
#6  black  purple     cat     2     0     1     0

Answer 2

Я сливаю df2 в df1 три раза рекурсивно по purrr::reduce(). В этой части я копирую df2 три раза и изменяю имена их первых столбцов, чтобы они соответствовали именам в df1 соответственно.

library(tidyverse)

df2 %>%
  list %>% rep(3) %>%
  imap(~ setNames(.x, c(str_c("group", .y), "Level", str_c("Score", .y)))) %>%
  reduce(left_join, .init = df1)

#   group1  group2  group3 Level Score1 Score2 Score3
# 1    cat     cat     dog     1      1      1      1
# 2    dog  parrot     cat     1      1      1      1
# 3  mouse dolphin dolphin     1      1      0      0
# 4    red    blue    blue     2      1      1      1
# 5  green  yellow   green     2      1      1      1
# 6  black  purple     cat     2      0      1      0