的value
不需要任何enquo
,因爲它是一個數字元素,而不是帶引號的字符串。另外,它不是一個專欄。它可以內mutate/summarise
make_dummy_at_value <- function(x, var, value) {
var_e <- enquo(var)
if (is.na(value)) {
x %>%
mutate(dum = as.numeric(is.na(!!var_e)))
} else {
x %>%
mutate(dum = as.numeric((!!var_e) == value & !is.na(!!var_e)))
}
}
make_dummy_at_value(x, some_var, 2)
# some_var dum
#1 NA 0
#2 2 1
make_dummy_at_value(x, some_var, NA)
# some_var dum
#1 NA 1
#2 2 0
make_dummy_at_value(x, some_var, 3)
# some_var dum
#1 NA 0
#2 2 0
直接使用如果目的是通過數字以及字符串「價值」,然後
make_dummy_at_value <- function(x, var, value) {
var_e <- enquo(var)
value <- enquo(value)
value_e <- quo_name(value)
if (is.na(value_e)) {
x %>%
mutate(dum = as.numeric(is.na(!!var_e)))
} else {
x %>%
mutate(dum = as.numeric((!!var_e) == value_e & !is.na(!!var_e)))
}
}
make_dummy_at_value(x, some_var, 2)
# some_var dum
#1 NA 0
#2 2 1
make_dummy_at_value(x, some_var, NA)
# some_var dum
#1 NA 1
#2 2 0
make_dummy_at_value(x, some_var, v1)
# some_var dum
#1 NA 0
#2 2 0
使用另一示例
x1 <- data.frame(col1 = c(NA, 4, 5), col2 = c('A', NA, 'B'), stringsAsFactors= FALSE)
make_dummy_at_value(x1, col1, NA)
# col1 col2 dum
#1 NA A 1
#2 4 <NA> 0
#3 5 B 0
make_dummy_at_value(x1, col2, NA)
# col1 col2 dum
#1 NA A 0
#2 4 <NA> 1
#3 5 B 0
make_dummy_at_value(x1, col2, A)
# col1 col2 dum
#1 NA A 1
#2 4 <NA> 0
#3 5 B 0
您是否想要在第5行寫上if(is.na(!! value_e))? – FlorianGD