Создайте новый столбец, используя mutate_at () в R

Question

Я пытаюсь внести некоторые изменения в следующий фрейм данных:

df <- data.frame(
    zgen = c("100003446", "100001749","100002644","100001755"),
    Name_mat = c("EVEROLIMUS 10 MG CM", "GALSULFASA 5MG/5ML FAM", "IDURSULFASE 2MG/ML SOL. P/INFUSION FAM","IMIGLUCERASA 400U POL. LIOF. FAM"),
    details= c("CM", "FAM", "SOL. P/INFUSION FAM","NA")
)

И я использую mutate_at() из пакета dplyr, чтобы создать новый столбец с именем "type". Этот столбец может меняться в зависимости от списка символов, которые могут отображаться в столбцах моего фрейма данных («name_mat» и «details»). Код:

df <- df %>% mutate_at(vars(one_of("Name_mat ","details")),
                       funs(case_when( "FAM|FRA" == TRUE ~ "FA",
                                       "CM|COMPRIMIDO" == TRUE~ "COM",
                                       "SOL"== TRUE~"SOL",
                                       "CP|CAPSULA"== TRUE~"CAP",
                                       TRUE ~ "bad_mat")))

Я впервые использую mutate_at, и я не знаю, как создать новый столбец с именем "type" в моем фрейме данных "df". Наконец, мне нужно что-то вроде:

       ZGEN                                 Name_mat               details   Type
1 100003446                      EVEROLIMUS 10 MG CM                    CM    COM
2 100001749                   GALSULFASA 5MG/5ML FAM                   FAM     FA
3 100002644   IDURSULFASE 2MG/ML SOL. P/INFUSION FAM   SOL. P/INFUSION FAM     FA
4 100001755         IMIGLUCERASA 400U POL. LIOF. FAM                    NA     FA

Я ценю любую помощь или любую другую точку зрения о том, как это сделать.

Спасибо!

Answer 1

Мы также можем использовать

library(dplyr)
library(purrr)
library(stringr)
pat <- "\\b(FAM|FRA|CM|COMPRIMIDO|SOL|CP|CAPSULA)\\b"

nm1 <- setNames(c("FA", "FA", "COM", "COM", "SOL", "CAP", "CAP"),
       c("FAM", "FRA", "CM", "COMPRIMIDO", "SOL", "CP", "CAPSULA"))
df %>% 
     select(Name_mat, details) %>%
     map(str_extract_all, pattern = pat) %>% 
           transpose %>% 
     map_chr( ~ nm1[flatten_chr(.x)][1] ) %>%
     bind_cols(df, Type = .)

Answer 2

попробуй сделать это так

  library(tidyverse)
  library(stringr)

  df %>% mutate(TYPE = case_when(
  str_detect(Name_mat, pattern = "FAM") | str_detect(details, "FRA") ~ "FA",
  str_detect(Name_mat, pattern = "CM") | str_detect(details, "COMPRIMODO") ~ "CM",
  str_detect(Name_mat, pattern = "SOL") ~ "SOL",
  str_detect(Name_mat, pattern = "CP") | str_detect(details, "CAPSULA") ~ "CAP",
  TRUE ~ "bad_mat"))