如何创建仅在直接调用时才打印内容的R函数？

解决方案：利用S3类自定义打印行为 #

要实现类似ggplot或lm的函数行为，核心是通过S3类自定义print方法，让函数返回的对象仅在顶层直接调用、被print调用时触发额外内容的打印，赋值时不执行打印。

代码实现 #

# 定义主函数，返回带自定义类的结果
my_fun <- function(x) {
  result <- x + 1
  # 为结果添加自定义类标识
  class(result) <- "my_fun_output"
  # 用invisible阻止自动打印返回值
  invisible(result)
}

# 自定义对应类的print方法
print.my_fun_output <- function(x, ...) {
  # 打印需要的额外内容
  print(0)
  # 若不需要打印返回值，直接用invisible返回；若需要打印，替换为NextMethod()
  invisible(x)
}

测试验证 #

my_fun(1)         #> 0    符合预期：顶层调用打印0，返回值不显示
print(my_fun(1))  #> 0    符合预期：仅打印0，返回值不显示
val <- my_fun(1)  #> 无输出 符合预期：赋值时不打印任何内容

# 查看赋值后的结果（需移除自定义类）
unclass(val)      #> 2

原理说明 #

1. 自定义类标识：给函数返回的结果添加专属类my_fun_output，让R识别该对象需要使用自定义的print方法。
2. invisible返回：阻止函数在顶层调用时自动打印原始返回值，仅触发自定义print方法的内容。
3. 自定义print方法：仅当对象被直接调用或print包裹时，执行额外内容的打印；赋值操作不会触发print方法，因此无输出。

如果需要在print调用时同时打印返回值，只需修改print方法为：

print.my_fun_output <- function(x, ...) {
  print(0)
  # 调用默认print方法打印返回值
  NextMethod()
}

内容的提问来源于stack exchange，提问作者vet.szabolcs