R语言中DataFrame列名作为函数参数
R语言中DataFrame列名作为函数参数
卡尔曼和玻尔兹曼谁曼
发布于 2022-10-04 21:17:03
发布于 2022-10-04 21:17:03
R语言中DataFrame列名作为函数参数
在使用Tidyverse提供的各种函数时，我们很多时候都会直接传递DataFrame的列名作为函数参数，对对应的列进行操作。如果我们自定义的函数中需要传递列名作为函数参数，如何实现呢？
直接传递列名会怎么样？
比如，我们自定义一个函数，用于计算指定列名的平均值
# 函数定义
col_mean <- function(data, colname) { data %>% summarise(across(colname, mean)) }
# 函数调用，这里的mpg为数据表中的列名
data <- read_csv(readr_example('mtcars.csv'))
col_mean(data, mpg)执行结果：
DataFrame列名作为函数参数
可以发现，直接使用列名传递到函数中会报错，下面介绍两种解决方案：
使用{{}}语法糖
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()函数将列名转为表达式，然后在使用的时候通过!!符号进行提取即可。
DataFrame列名作为函数参数
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2022-09-30，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录