R/words_replace.R

In simonedambrogio/user:

#The words_rm function eliminates the worlds that we want to eliminate
#There are only 4 arguments:
#data: name of the dataset
#variable: character variable that we want to modify. don't forget the quotation marks
#words_to_rm: words that we want to eliminate separated by |
#replace_with: numbers or words we want instead of words_rm. Default is "" and you should 
#keep it if you want to eliminate the words

words_replace <- function (variable, words_to_replace, replace_with = "") {
  
  
  library(dplyr)
  if (!require(dplyr)) {
    install.packages("dplyr")
    library(dplyr)
  }
  
  # variable <- enquo(variable)
  # var <- select(data, !!variable) %>% .[,]
  
  if(!is.character(variable)) variable <- as.character(variable)
  
  for (word_i in 1:length(words_to_replace)) {
    word_to_rm_index <- which(str_detect(variable, fixed(words_to_replace[word_i])))
    
    if(any(word_to_rm_index)) {
      words_to_replace_i <- words_to_replace[word_i]
      new_words <- sapply(seq_along(word_to_rm_index), function(i){
        variable[word_to_rm_index][i] <- gsub(words_to_replace_i, replace_with, variable[word_to_rm_index][i], fixed = TRUE)
      })
      variable[word_to_rm_index] <- new_words
    } else {
      warning( paste0("Sting '",  words_to_replace ,"' not found") )
    }
  }
  
  return(variable)
}




#words_rm(data = dec, variable = value_up_boundary, words_to_rm = "\\+")