R重命名函数中传递的列

Question

我一直在搜索这个并发现this link 有助于从函数重命名传递的列（[,column_name] 代码实际上使 my_function1 在我搜索了一段时间后工作。有没有办法使用管道运算符重命名函数内数据框中的列？

我的尝试显示在my_function2 中，但它给了我Error: All arguments to rename must be named 或Error: Unknown variables: col2。我猜是因为我没有指定col2 属于什么。

另外，有没有办法将关联的参数传递给函数，例如 col1 和 new_col1，以便您可以关联要替换的列名和替换它的列名。提前致谢！

library(dplyr)

my_df = data.frame(a = c(1,2,3), b = c(4,5,6), c = c(7,8,9))

my_function1 = function(input_df, col1, new_col1) {
  df_new = input_df
  df_new[,new_col1] = df_new[,col1]
  return(df_new)
}
temp1 = my_function1(my_df, "a", "new_a")

my_function2 = function(input_df, col2, new_col2) {
  df_new = input_df %>%
    rename(new_col2 = col2)
  return(df_new)
}

temp2 = my_function2(my_df, "b", "new_b")

Answer 1

rename_（与其他带有下划线后缀的 dyplyr 动词一起）已被贬值。 相反，请尝试：

my_function3 = function(input_df, cols, new_cols) { 
  input_df %>%
    rename({{ new_cols }} := {{ cols }}) 
}

请参阅this vignette，了解有关使用双括号包含参数和使用 dplyr 编程的更多信息。

Answer 2

按照@MatthewPlourde's answer to a similar question，我们可以这样做：

my_function3 = function(input_df, cols, new_cols) { 
  rename_(input_df, .dots = setNames(cols, new_cols)) 
}

# example
my_function3(my_df, "b", "new_b")
#   a new_b c
# 1 1     4 7
# 2 2     5 8
# 3 3     6 9

许多 dplyr 函数都有名称以_ 结尾的鲜为人知的变体。这允许您以更编程的方式使用包。一种模式是...

DF %>% dplyr_fun(arg1 = val1, arg2 = val2, ...)
# becomes
DF %>% dplyr_fun_(.dots = list(arg1 = "val1", arg2 = "val2", ...))

这在少数情况下对我有用，其中 val* 只是列名。还有更复杂的模式和技巧，在你输入vignette("nse")时弹出的文档里有介绍，不过我不是很了解。