I'm trying to recode all but one type of element in a column inside a function. The column name changes depending on the input of the function, and also the element in the column which shouldn't be recoded changes depending on the function input.

the data frame (df in code snippet) looks as follows:

but could also look like

The output that I want is the following respectively:

I have tried the following:

select_data <- function(df, col_name, do_not_change){
     df <- df %>%
       dplyr::mutate('{col_name}' = recode(col_name, '{do_not_change}' := {{do_not_change}}, .default = "rest"))

So the calls would look as follows respectively:

# case of education
education_df <- function(df, education, 'bachelor')
# case of gender
gender_df <- function(df, gender, 'male')

However, this doesn't seem to be working. Any help would be greatly appreciated. Thanks a ton in advance!

CodePudding user response：

The function can be modified to

select_data <- function(df, col_name, do_not_change){
     df %>%
       dplyr::mutate({{col_name}} := recode({{col_name}}, 
           {{do_not_change}} := {{do_not_change}}, .default = "rest"))
      
       }

and call as

select_data(df, gender, 'male')  
  gender   education
1   male    bachelor
2   rest      master
3   rest high school
4   male      master 

data

df <- structure(list(gender = c("male", "female", "female", "male"), 
    education = c("bachelor", "master", "high school", "master"
    )), row.names = c(NA, -4L), class = "data.frame")

CodePudding user response：

Would this work ?

df %>%
  mutate(
    gender = ifelse(gender == "male", "male", "rest"),
    education = ifelse(education == "bachelor", "bachelor", "rest"))