Início / coding / Perguntas / 77228204
Accepted
WalliYo_
Asked: 2023-10-04 17:04:33 +0800 CST

Mutação dinâmica case_when de acordo com a correspondência de string

  • 772

Eu tenho um vetor que armazena todas as combinações possíveis de mesclagem para uma variável qualitativa.
O vetor tem a seguinte estrutura:

Config = c("4 With 1 + 6 With 2 + 8 With 3","4 With 1 With 2 + 6 With 8")

O que significa que a variável com o primeiro valor de Config deve ter 3 modalidades: Valores 4 com 1, Valores 6 com 2, Valores 8 com 3

Gostaria de atribuir um valor dinâmico a cada elemento dessa fusão de forma dinâmica, como:

mtcars %>% mutate(Merging=case_when(carb %in% c(4,1)~"Group 1", carb %in% c(6,2)~"Group 2",carb %in% c(8,3) ~"Group 3"))

Minha principal dificuldade aqui é que o número de grupos não é o mesmo para cada elemento da configuração: para uma configuração podem ser 3 grupos, para outra 4, etc.

Já tentei trabalhar com string match mas essa possibilidade só funciona para 1 grupo que contém apenas a palavra "With" :

mutate(Merging= case_when(
      !!sym(VAR) %in% c(unlist(str_split(Config, " With "))) ~ "Group 1",
      TRUE ~ !!sym(VAR)
    ))

É uma maneira de fazer isso dinamicamente para cada elemento do vetor e criar a variável dedicada a cada iteração?

Muito obrigado

1 respostas

  1. Best Answer

    está tudo bem

    library(dplyr)
library(stringr)
library(purrr)
library(rlang)

# Sample data
Config <- c("4 With 1 + 6 With 2 + 8 With 3", "4 With 1 With 2 + 6 With 8")
mtcars <- data.frame(carb = c(4, 1, 2, 6, 8, 3)) # Sample data

# Function to create dynamic conditions
create_conditions <- function(config) {
  pairs <- str_split(config, "\\s*\\+\\s*")[[1]]
  
  conditions <- map2(pairs, 1:length(pairs), function(pair, group_num) {
    values <- as.numeric(str_extract_all(pair, "\\d+")[[1]])
    group_name <- paste("Group", group_num)
    condition <- expr(carb %in% c(!!!values) ~ !!group_name)
    return(condition)
  })
  
  return(conditions)
}

# Create a list of conditions for each element of Config
conditions_list <- map(Config, create_conditions)

# Create a dynamic column using case_when
mtcars <- mtcars %>%
  mutate(Merging = case_when(!!!set_names(unlist(conditions_list), NULL)))

# View the result
print(mtcars)
    • 1

relate perguntas