带有 rename_with 的 Purrr 地图

Question

我正在尝试清理数据集的名称。我用janitor::clean_names() 开始。但是，我仍然有一些缩写，我想用下划线 _ 分开。我有使用rename_with(~str_replace(.x, "gh", "gh_"), .cols = starts_with("gh")) 的代码，但是有很多缩写，最好找到map 的方法或以其他方式功能化此过程。

dat <- tibble(ghrisk_value = c(1,2), 
              ghrisk_corrected = c(2,3), 
              devpolicy_value = c(4,5),
              devpolicy_corrected = c(5,6))

# code works but not functionalized
dat %>%
   rename_with(~str_replace(.x, "gh", "gh_"), .cols = starts_with("gh")) %>%
   rename_with(~str_replace(.x, "dev", "dev_"), .cols = starts_with("dev")) %>%
   names()

# attempt at map...
abbr_words <- c("gh", "dev")
map(dat, ~rename_with(str_replace(.x, abbr_words, str_c(abbr_words, "_"))) 

Answer 1

使用地图，您将需要一个辅助函数，即 real_func。 map 将适用于 colnames(dat)，并且一次只能使用一个 colname。 Map需要一个函数real_func，第一个参数，即数据参数将在函数之前，其余参数将在函数之后。 Repl_func 将一次获取一个列名并获取缩写词列表，循环遍历它并执行替换。在 unlist 结束时，需要返回一个扁平化的向量。

abbr_words <- c("gh", "dev")

repl_func <- function(x,y){
  for (i in y){
    x <- str_replace(x,i,paste0(i,"_"))
  }
  return (x)
}


colnames(dat) <- unlist(map(colnames(dat), repl_func, abbr_words))

Answer 2

您可以将reduce 上面的单词替换为str_replace

abbr_words <- c("gh", "dev")

dat %>% 
  rename_all( ~
    reduce(abbr_words, ~str_replace(.x, paste0('^', .y), paste0(.y, '_')), .init = names(dat))
  )

# # A tibble: 2 x 4
#   gh_risk_value gh_risk_corrected dev_policy_value dev_policy_corrected
#           <dbl>             <dbl>            <dbl>                <dbl>
# 1             1                 2                4                    5
# 2             2                 3                5                    6

Answer 3

您不需要map()。只需使用正则表达式语法"(?<=a|b|c)"，它匹配a 或b 或c 后面的位置并插入下划线。另外starts_with()可以将一个字符向量作为输入来匹配所有元素的并集。

abbr_words <- c("gh", "dev")

pattern <- sprintf("(?<=%s)", str_c(abbr_words, collapse = "|"))
# [1] "(?<=gh|dev)"

dat %>%
  rename_with(~ str_replace(.x, pattern, "_"), starts_with(abbr_words))

# # A tibble: 2 x 4
#   gh_risk_value gh_risk_corrected dev_policy_value dev_policy_corrected
#           <dbl>             <dbl>            <dbl>                <dbl>
# 1             1                 2                4                    5
# 2             2                 3                5                    6