В R l oop просматривают список переменных и создают новые условные значения переменных

Question

Я пытаюсь создать новую переменную (категорию) всякий раз, когда любой из 12 других переменных (от var1 до var12) удовлетворяет одному из многих условий:

for (i in 1:12) {
  new_df <- df %>% mutate(
    category = case_when(df[paste0("var",i)] == "09F" ~ "arts",
                         df[paste0("var",i)] == "09O" ~ "arts_o",
                         TRUE ~ " "))
}

, но переменная категории по-прежнему пуста ... Кто-нибудь знает, почему этот код не работает? Должен ли я использовать любой другой метод?

Спасибо!

Answer 1

Используя некоторые случайные данные примера только с переменными (var1 и var2, но могут быть обобщены для вашего случая), проблема присвоения категории каждому obs в зависимости от значений var1 и var2 может быть решена следующим образом:

set.seed(42)

# Example data
df <- data.frame(
  id = 1:10,
  var1 = sample(c("09F", "09O", "10F"), 10, replace = TRUE),
  var2 = sample(c("09F", "09O", "10F"), 10, replace = TRUE),
  stringsAsFactors = FALSE
)

library(tidyr)
library(dplyr)

df %>% 
  # Convert to long
  pivot_longer(var1:var2) %>% 
  # Assign category
  group_by(id) %>% 
  mutate(category = case_when(
    any(value == "09F") ~ "arts",
    any(value == "09O") ~ "arts_o",
    TRUE ~ " ")) %>% 
  ungroup() %>% 
  # Convert back to wide
  pivot_wider(id_cols = c(id, category), names_from = "name", values_from = "value")
#> # A tibble: 10 x 4
#>       id category var1  var2 
#>    <int> <chr>    <chr> <chr>
#>  1     1 "arts"   09F   09F  
#>  2     2 "arts"   09F   09F  
#>  3     3 "arts"   09F   09O  
#>  4     4 "arts"   09F   09O  
#>  5     5 "arts_o" 09O   09O  
#>  6     6 "arts_o" 09O   10F  
#>  7     7 "arts_o" 09O   10F  
#>  8     8 "arts"   09F   09F  
#>  9     9 "arts"   10F   09F  
#> 10    10 " "      10F   10F

^{Создано в 2020-04-15 пакетом Представления (v0.3.0)}

Answer 2

Tidy подход:

set.seed(234)
sample(c("09F", "09O", "xyz"), 12 * 10, replace = TRUE, prob = c(0.05, 0.05, 0.9)) %>%
  matrix(ncol = 12) %>%
  `colnames<-`(str_c("var", 1:12)) %>%
  as_tibble() -> d

d %>%
  mutate(category = map_chr(
    pmap(d %>%
           select(starts_with("var")),
         c),
    ~case_when(any(.x == "09F") ~ "arts",
               any(.x == "09O") ~ "arts_O",
               TRUE ~ " ")))

## A tibble: 10 x 13
#   var1  var2  var3  var4  var5  var6  var7  var8  var9  var10 var11 var12 category
#   <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr> <chr>   
# 1 xyz   xyz   xyz   xyz   xyz   xyz   xyz   09O   xyz   xyz   xyz   xyz   "arts_O"
# 2 xyz   xyz   xyz   xyz   09O   xyz   xyz   xyz   xyz   xyz   xyz   xyz   "arts_O"
# 3 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   " "     
# 4 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   " "     
# 5 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   09O   xyz   xyz   xyz   "arts_O"
# 6 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   09F   xyz   xyz   "arts"  
# 7 09O   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   "arts_O"
# 8 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   09F   xyz   xyz   xyz   "arts"  
# 9 09O   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   "arts_O"
#10 xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   xyz   " "    

Или база R:

d$category2 <- ifelse(apply(d[grepl("var", names(d))], 1, function(x) any(x == "09F")), "arts", 
                      ifelse(apply(d[grepl("var", names(d))], 1, function(x) any(x == "09O")), "arts_o", " "))    

Answer 3

Мы можем использовать mutate_at для применения функции к нескольким столбцам вместо for l oop. Попробуйте:

library(dplyr)

new_df <- df %>% 
    mutate_at(vars(var1:var12), ~list(category = case_when(. == "09F" ~ "arts",
                                                           . == "09O" ~ "arts_o",
                                                           TRUE ~ " ")))