如何适配自定义tab函数以支持R语言原生管道运算符|>？

想要开发tab(x, y)或tab(x, y, d)（其中d为data.frame）函数，替代以下命令：

d <- mtcars
d |> dplyr::select(cyl, vs) |> table()

尝试编写的函数如下：

d <- mtcars
tab <- function (x, y) {
  result <- table(x, y)
  result
}

该函数在不使用管道时可正常运行：

tab(d$cyl, d$vs)

但使用R原生管道运算符|>时无法运行：

d |> tab(cyl, vs)

而使用magrittr包中的曝光管道%$%可以正常运行：

d %$% tab(cyl, vs)

需要修改函数以适配R原生管道运算符|>。

R原生管道|>的核心逻辑是将左侧对象作为第一个参数传入右侧函数，你当前的tab函数第一个参数是列向量而非数据框，直接用管道会把整个d传给x，导致报错。以下是几种适配原生管道的可行方案：

方案1：调整参数顺序（推荐） #

修改函数，让第一个参数接收数据框，内部通过非标准求值提取指定列：

tab <- function(data, x, y) {
  x_col <- data[[deparse(substitute(x))]]
  y_col <- data[[deparse(substitute(y))]]
  table(x_col, y_col)
}

此时原生管道调用可正常工作：

d <- mtcars
d |> tab(cyl, vs)

若要直接传向量，只需显式指定参数名：

tab(data = NULL, x = d$cyl, y = d$vs)

方案2：兼容多种调用格式 #

如果希望同时支持直接传向量、管道传数据框、显式传数据框三种调用方式，可以通过参数类型判断实现：

tab <- function(x, y, d = NULL) {
  # 处理管道调用：左侧数据框作为第一个参数传入
  if (is.data.frame(x)) {
    d <- x
    x_sym <- substitute(y)
    y_sym <- substitute(d)
    x_col <- d[[deparse(x_sym)]]
    y_col <- d[[deparse(y_sym)]]
  } else if (!is.null(d)) {
    # 处理显式传入数据框的情况
    x_col <- d[[deparse(substitute(x))]]
    y_col <- d[[deparse(substitute(y))]]
  } else {
    # 处理直接传向量的情况
    x_col <- x
    y_col <- y
  }
  table(x_col, y_col)
}

三种调用方式均有效：

# 直接传向量
tab(d$cyl, d$vs)
# 原生管道调用
d |> tab(cyl, vs)
# 显式指定数据框
tab(cyl, vs, d = d)

方案3：用rlang简化语法（需依赖rlang包） #

若愿意依赖rlang包，可使用{{}}语法简化非标准求值代码：

library(rlang)
tab <- function(data, x, y) {
  table({{x}}, {{y}}, data = data)
}

调用方式与方案1一致：

d |> tab(cyl, vs)

内容的提问来源于stack exchange，提问作者falafel