在 R 中，根据独占条件创建多个新变量

Question

我需要帮助完成一项看似简单的任务。我想基于< code>dplyr::mutate中的一个条件创建几个新变量。我可以使用< code>ifelse创建一个新变量，但是我想一步创建几个。

让我们假设这是我的数据帧。

df_have <- data.frame(id = 1:10, x = 1:2, y = sample(10, replace = T))

我想要这样的东西:

df_want <- mutate(df_have,
                  if_else(y<3, var1 = x, var2 = y/x, var3 = T)) 

因此，if条件应该基于条件< code>y创建三个新变量

我想与< code>if_else和< code>case_when一起使用。

谢谢转发

Answer 1

像这样一次创建所有的变量，尽管坦白地说我不推荐这样做！：

func <- function(x,y) {
  if(y<3) list("var1"=x,"var2"=y/x,"var3"=T)
  else(list("var1"=NA, "var2"=NA, "var3"=NA))
}
df_have %>%
  rowwise() %>% 
  mutate(vars=list(func(x,y))) %>%
  unnest_wider(col = vars)

输出：

      id     x     y  var1  var2 var3 
   <int> <int> <int> <int> <dbl> <lgl>
 1     1     1     7    NA    NA NA   
 2     2     2     2     2     1 TRUE 
 3     3     1     5    NA    NA NA   
 4     4     2     3    NA    NA NA   
 5     5     1     1     1     1 TRUE 
 6     6     2     3    NA    NA NA   
 7     7     1     5    NA    NA NA   
 8     8     2     4    NA    NA NA   
 9     9     1     2     1     2 TRUE 
10    10     2     4    NA    NA NA  

这样做要快得多：

df_have %>% 
  mutate(
    var1=if_else(y<3, x, as.integer(NA)),
    var2=if_else(y<3, y/x, as.double(NA)),
    var3 =if_else(y<3, T, as.logical(NA))
  )

Answer 2

您可以在mutate中为每个新变量单独使用< code>case_when。请注意，所有< code>case_when案例的输出需要是相同的类型，因此< code>NA_？？_类型可能需要根据实际数据进行调整。

set.seed(20)
df_have <- data.frame(id = 1:10, x = 1:2, y = sample(10, replace = T))

df_have %>%
  dplyr::mutate(
    var1 = case_when(y < 3 ~ x,
                     TRUE ~ NA_integer_),
    var2 = case_when(y < 3 ~ y/x,
                     TRUE ~ NA_real_),
    var3 = case_when(y < 3 ~ TRUE,
                     TRUE ~ FALSE)
  )

#   id x y var1 var2  var3
#1   1 1 6   NA   NA FALSE
#2   2 2 8   NA   NA FALSE
#3   3 1 2    1    2  TRUE
#4   4 2 9   NA   NA FALSE
#5   5 1 2    1    2  TRUE
#6   6 2 9   NA   NA FALSE
#7   7 1 3   NA   NA FALSE
#8   8 2 9   NA   NA FALSE
#9   9 1 8   NA   NA FALSE
#10 10 2 2    2    1  TRUE