在`DPLYR`中使用动态变量名

Question

我想使用dplyr::mutate()在数据帧中创建多个新列。列名及其内容应动态生成。

来自IRIS的示例数据：

library(dplyr)
iris <- as_tibble(iris)

我创建了一个函数来从petal.width变量中更改新列：

multipetal <- function(df, n) {
    varname <- paste("petal", n , sep=".")
    df <- mutate(df, varname = Petal.Width * n)  ## problem arises here
    df
}

for(i in 2:5) {
    iris <- multipetal(df=iris, n=i)
}

Answer 1

由于您要动态地将变量名构建为字符值，因此使用标准data.frame索引进行赋值更有意义，该索引允许为列名设置字符值。例如：

multipetal <- function(df, n) {
    varname <- paste("petal", n , sep=".")
    df[[varname]] <- with(df, Petal.Width * n)
    df
}

mutate函数使通过命名参数命名新列变得非常容易。但前提是您在键入命令时知道名称。如果要动态指定列名，那么还需要构建命名参数。

使用最新的dplyr版本，在使用:=命名参数时，可以使用glue包中的语法。因此，在这里，名称中的{}通过计算内部的表达式来获取值。

multipetal <- function(df, n) {
  mutate(df, "petal.{n}" := Petal.Width * n)
}

# --- dplyr version 0.7+---
multipetal <- function(df, n) {
    varname <- paste("petal", n , sep=".")
    mutate(df, !!varname := Petal.Width * n)
}

有关更多信息，请参阅vignette（“programming”,“dplyr”）格式的可用文档。

dplyr稍早的版本(>=0.3<0.7)，鼓励使用“标准评估”替代许多函数。有关更多信息，请参阅非标准评估小插图(小插图（“NSE”）)。

所以在这里，答案是使用mutate_()而不是mutate()并执行：

# --- dplyr version 0.3-0.5---
multipetal <- function(df, n) {
    varname <- paste("petal", n , sep=".")
    varval <- lazyeval::interp(~Petal.Width * n, n=n)
    mutate_(df, .dots= setNames(list(varval), varname))
}

# --- dplyr versions < 0.3 ---
multipetal <- function(df, n) {
    varname <- paste("petal", n , sep=".")
    pp <- c(quote(df), setNames(list(quote(Petal.Width * n)), varname))
    do.call("mutate", pp)
}