[英]Using a custom case_when function inside a mutate call in dplyr issues
[英]Writing a custom case_when function to use in dplyr mutate using tidyeval
我正在嘗試編寫一個自定義 case_when 函數以在 dplyr 中使用。 我一直在閱讀其他問題中發布的 tidyeval 示例,但仍然無法弄清楚如何使其工作。 這是一個reprex:
df1 <- data.frame(animal_1 = c("Horse", "Pig", "Chicken", "Cow", "Sheep"),
animal_2 = c(NA, NA, "Horse", "Sheep", "Chicken"))
translate_title <- function(data, input_col, output_col) {
mutate(data,
!!output_col := case_when(
input_col == "Horse" ~ "Cheval",
input_col == "Pig" ~ "Рorc",
input_col == "Chicken" ~ "Poulet",
TRUE ~ NA)
)
}
df1 %>%
translate_title("animal_1", "animaux_1") %>%
translate_title("animal_2", "animaux_2")
當我嘗試運行它時,我Error in mutate_impl(.data, dots) : Evaluation error: must be type string, not logical.
遇到Error in mutate_impl(.data, dots) : Evaluation error: must be type string, not logical.
此外,我實際上想重寫該函數,以便可以像這樣使用它:
df1 %>%
mutate(animaux_1 = translate_title(animal_1),
animaux_2 = translate_title(animal_2)
)
但不確定如何。
根據您希望如何將輸入傳遞給函數,您可以通過兩種方式解決它:
1) 使用{{}}
以不帶引號的形式傳遞輸入
library(dplyr)
translate_title <- function(data, input_col, output_col) {
mutate(data,
!!output_col := case_when(
{{input_col}} == "Horse" ~ "Cheval",
{{input_col}} == "Pig" ~ "Рorc",
{{input_col}} == "Chicken" ~ "Poulet",
TRUE ~ NA_character_)
)
}
df1 %>%
translate_title(animal_1, "animaux_1") %>%
translate_title(animal_2, "animaux_2")
# animal_1 animal_2 animaux_1 animaux_2
#1 Horse <NA> Cheval <NA>
#2 Pig <NA> Рorc <NA>
#3 Chicken Horse Poulet Cheval
#4 Cow Sheep <NA> <NA>
#5 Sheep Chicken <NA> Poulet
2) 傳遞使用sym
和!!
引用的輸入
translate_title <- function(data, input_col, output_col) {
mutate(data,
!!output_col := case_when(
!!sym(input_col) == "Horse" ~ "Cheval",
!!sym(input_col) == "Pig" ~ "Рorc",
!!sym(input_col) == "Chicken" ~ "Poulet",
TRUE ~ NA_character_)
)
}
df1 %>%
translate_title("animal_1", "animaux_1") %>%
translate_title("animal_2", "animaux_2")
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.