R语言中DataFrame列名作为函数参数

Posted 2022-10-04 卡尔曼和玻尔兹曼谁曼

R语言中DataFrame列名作为函数参数

• 直接传递列名会怎么样？
• 使用语法糖
• 使用enquo函数和!!语法糖

在使用Tidyverse提供的各种函数时，我们很多时候都会直接传递DataFrame的列名作为函数参数，对对应的列进行操作。如果我们自定义的函数中需要传递列名作为函数参数，如何实现呢？

直接传递列名会怎么样？

比如，我们自定义一个函数，用于计算指定列名的平均值

# 函数定义
col_mean <- function(data, colname)  data %>% summarise(across(colname, mean)) 
# 函数调用，这里的mpg为数据表中的列名
data <- read_csv(readr_example('mtcars.csv'))
col_mean(data, mpg)

执行结果：

可以发现，直接使用列名传递到函数中会报错，下面介绍两种解决方案：

使用语法糖

col_mean <- function(data, colname)  data %>% summarise(across(colname, mean)) 

在函数体中使用将列名括起来即可。

使用enquo函数和!!语法糖

col_mean <- function(data, colname)  colname <- enquo(colname); data %>% summarise(across(!!colname, mean)) 

第二种方法是在函数体内部，先使用enquo()函数将列名转为表达式，然后在使用的时候通过!!符号进行提取即可。