如何在 R 中编写 function 以接受 dplyr 等列名？

Question

我正在编写一个 package ，其中有几个函数接受 dataframe object 以及数据帧的列名 297DBC11FBZABDEDA8 。

这是一个简化的示例：

func = function(df,vars){
    head(df[,vars])
}

#column args as strings
func(mtcars,c("mpg","cyl"))

我不希望将列名作为字符串提供，而是希望 function 接受（并建议/自动完成）像 dplyr 函数中的列名。

#dplyr-style args
func(mtcars, mpg, cyl)

#which doesnt work because mpg and cyl don't exist as objects

我考虑使用...作为 function arguments 但这仍然涉及使用字符串。

任何帮助，将不胜感激。

Answer 1

一个可能的解决方案，使用dplyr ：

library(dplyr)

func = function(df,...){
  df %>% 
    select(...) %>% 
    head
}


func(mtcars, mpg, cyl)
#>                    mpg cyl
#> Mazda RX4         21.0   6
#> Mazda RX4 Wag     21.0   6
#> Datsun 710        22.8   4
#> Hornet 4 Drive    21.4   6
#> Hornet Sportabout 18.7   8
#> Valiant           18.1   6

func(mtcars, mpg)

#>                    mpg
#> Mazda RX4         21.0
#> Mazda RX4 Wag     21.0
#> Datsun 710        22.8
#> Hornet 4 Drive    21.4
#> Hornet Sportabout 18.7
#> Valiant           18.1

---

或者在base R中：

func = function(df,...){
  head(df[, sapply(substitute(...()), deparse)])
}

func(mtcars, mpg, cyl)
#>                    mpg cyl
#> Mazda RX4         21.0   6
#> Mazda RX4 Wag     21.0   6
#> Datsun 710        22.8   4
#> Hornet 4 Drive    21.4   6
#> Hornet Sportabout 18.7   8
#> Valiant           18.1   6

func(mtcars, mpg)

#> [1] 21.0 21.0 22.8 21.4 18.7 18.1

Answer 2

您可以使用

subset(df, select = item)

您应该查看Hadley Wickham 的 Advanced R，这非常有趣，如果有点，那么，高级。 尤其是：

20.4 数据掩码

在本节中，您将了解数据掩码，这是一个数据框，评估代码将首先在其中查找变量定义。 数据掩码是为 with()、subset() 和 transform() 等基本函数提供支持的关键思想，并在整个 tidyverse 中用于 dplyr 和 ggplot2 等包中。