使用“data”参数编写函数

我想编写一个函数，它可以将数据框中的列或列名称及其来自的数据帧作为参数。

df <- data.frame(x = c(1:5), y = c(6:10), z = LETTERS[1:5])

my_fxn <- function (aaa, bbb, ccc, data) {
  if (!missing(data)) {
    aaa = as.numeric(data$aaa)
  }
  print(aaa[1])
}

当我提供数据框中的列作为参数时，此函数工作正常：

> my_fxn(df$x, df$y, df$z, df)
[1] 1

如果列名称和数据框名称与函数参数的名称不同，则该方法不起作用：

> my_fxn(x, y, z, df)
[1] NA

但是，当列名称和数据框名称与函数参数的名称相同时，它确实有效：

data <- df
names(data) <- c("aaa", "bbb", "ccc")
> my_fxn(aaa, bbb, ccc, data)
[1] 1

到底是怎么回事？

Thanks!

借助 Cet - 您可能正在寻找这样的东西：

df <- data.frame(x = c(1:5), y = c(6:10), z = LETTERS[1:5])

my_fxn <- function (aaa, bbb, ccc, data) {
  if (!missing(data)) {
    aaa = as.numeric(data[[aaa]])
    bbb = as.numeric(data[[bbb]])
    ccc = as.character(data[[ccc]])
  }
  print(aaa[1])
}

my_fxn("x", "y", "z", df)
#> [1] 1

随着使用enquo() from library(dplyr)，我们不再需要输入字符作为函数变量：

library(dplyr)

my_fxn <- function (aaa, bbb, ccc, data) {
  aaa <- enquo(aaa)
  bbb <- enquo(bbb)
  ccc <- enquo(ccc)

  if (!missing(data)) {
    aaa = as.numeric(pull(data, !!aaa))
    bbb = as.numeric(pull(data, !!bbb))
    ccc = as.character(pull(data, !!ccc))
  }
  print(aaa[1])
}

my_fxn(x, y, z, df)
#> [1] 1

有关功能构建的更多信息enquo() and !!可以在这里找到：https://dplyr.tidyverse.org/articles/programming.html#programming-recipes

最后，使用基本 R 解决方案deparse() and substitute():

my_fxn <- function (aaa, bbb, ccc, data) {
  aaa <- deparse(substitute(aaa))
  bbb <- deparse(substitute(bbb))
  ccc <- deparse(substitute(ccc))

  if (!missing(data)) {
    aaa = as.numeric(data[[aaa]])
    bbb = as.numeric(data[[bbb]])
    ccc = as.character(data[[ccc]])
  }
  print(aaa[1])
}

my_fxn(x, y, z, df)
#> [1] 1