Добавление нового столбца в dataframe на основе нескольких условий двух других столбцов в R

Question

В кадре данных у меня есть два столбца A и B. Значения в каждом столбце могут быть либо 1, либо NA.
Я хочу добавить третий столбец с именем C, чтобы:

 - If A == NA AND B == NA, then C = "None"
 - If A == 1 AND B == NA, then C = "A" 
 - If A == NA AND B == 1, then C = "B"
 - if A == 1 AND B == 1, then C = "Both"

Answer 1

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

library(dplyr)
df %>%
    mutate(C = ifelse(is.na(A) & is.na(B), "None",
                   ifelse(A== 1 & is.na(B), "A",
                   ifelse(B==1 & is.na(A), "B",
                   "Both"))))

Answer 2

Просто переведите ваши условия в case_when заявления

library(dplyr)

df %>%
  mutate(C = case_when(is.na(A) & is.na(B) ~ "None", 
                       A == 1 & is.na(B) ~ "A",
                       B == 1 & is.na(A) ~ "B", 
                       TRUE ~ "Both"))