用于比较列的 R 函数

Question

在 R 语言中，我想创建一个函数来查看选定的列，以便在查看器中进行比较。假设我的数据帧是 df1:

compare_col <- function(x){
  select(df1, x) %>%
    View()
}

如果我用x定义函数，我只能输入1列。

compare_col <- function(x)

compare_col("col_1")

只有当我通过 x,y 定义函数时，我才能输入 2 列。

compare_col <- function(x, y)

compare_col("col_1", "col_2")

如何创建一个足够动态的函数来输入任何数字。列数？

Answer 1

您可以使用rlang包来实现这一点。这将允许您使用 syms 和 !!! 运算符输入一串列名称，该运算符将根据您的需要在给定环境中动态拼接和计算。

library(dplyr)
#library(rlang)
compare_col <- function(x){
  df1 %>% select(!!! syms(x)) %>%
    View()
}
compare_col(c("col1", "col2"))