首页 > 解决方案 > 使用 mutate 和 case_when 创建二分法 var (0/1) 的函数

问题描述

我正在尝试创建一个函数,它将类别为 1 / 2 的变量重新编码为 0 / 1。像这样:

var_dic <- function(data, x){
  mutate(x_rec, case_when(
              x == 1 ~ 1,
              x == 2 ~ 0,
              TRUE ~ NA_real_))
}

var1 <- c('1', '2', '1', '1', '2', NA, '2', NA)
var2 <- c('1', '1', '2', '2', '2', NA, '2', NA)
id <- c(1,2,3,4,5,6, 7, 8)
testing.df <- data.frame(id, var1, var2)

但我收到与“x_rec”相关的错误。我希望这部分代码建立新变量的名称。我猜我应该在某处包含 paste0,但不确定代码中的方式或位置。

标签: rfunctiondplyrcase-when

解决方案


我们也可以data使用

var_dic <- function(data, x){
   data %>%
      mutate(!! {{x}} := case_when(
          {{x}} == 1 ~ 1,
          {{x}} == 2 ~ 0,
          TRUE ~ NA_real_))
    }

推荐阅读