2

我正在尝试创建一个自定义函数,其中仅在数据框的一列上执行某些操作。但我希望该函数以这样一种方式工作,它不仅会输出执行操作的列,而且还会输出从中绘制特定列的整个数据框。这是我想要实现的一个非常简单的示例:

# libraries needed
library(dplyr)
library(rlang)

# creating a dataframe
data <-
  as.data.frame(cbind(
    x = rnorm(5),
    y = rnorm(5),
    z = rnorm(5)
  ))

# defining the custom function
custom.fn <- function(data, x) {
  df <- dplyr::select(.data = data,
                      x = !!rlang::enquo(x)) # how can I also retain all the other columns in the dataframe apart from x?
  df$x <- df$x / 2
  return(df)
}

# calling the function (also want y and z here in the ouput)
custom.fn(data = data, x = x)
#>             x
#> 1  0.49917536
#> 2 -0.03373202
#> 3 -1.24845349
#> 4 -0.15809688
#> 5  0.11237030

reprex 包(v0.1.1.9000)于 2018 年 2 月 14 日创建。

4

1 回答 1

2

只需指定要包含在select通话中的列:

custom.fn <- function(data, x) {
  df <- dplyr::select(.data = data,
                      x = !!rlang::enquo(x), y, z)
  df$x <- df$x / 2
  return(df)
}

如果您不想明确命名其余列,也可以使用everything

df <- dplyr::select(.data = data, x = !!rlang::enquo(x), dplyr::everything())
于 2018-02-14T22:53:14.560 回答