如何为自定义R函数的返回结果实现plot方法及新增参数？

当然可行！这正是R的S3方法系统擅长的场景——给自定义对象类型绑定专属的绘图逻辑，就像survival包对survfit对象做的那样。下面是具体的实现步骤和示例：

首先修改你的函数，让它返回的列表带上一个自定义类名，这样R才能识别它并调用对应的绘图方法：

f <- function(x, y){
  b <- x^2
  c <- y^2
  d <- x+y
  result <- list(one = b, two = c, three = d)
  class(result) <- "my_custom_list"  # 给列表添加自定义类标识
  return(result)
}

接下来我们写一个plot.my_custom_list函数（命名规则是plot.你的类名），定义默认绘图逻辑，同时添加自定义参数。这里我们新增一个show_three参数（TRUE/FALSE），控制是否添加和three向量相关的线条：

plot.my_custom_list <- function(x, show_three = FALSE, ...){
  # 默认绘制one vs two的图形，和plot(dat$one, dat$two)效果一致
  plot(x$one, x$two, ...)  # ...用来传递plot函数的原生参数，比如xlab、ylab、pch等
  
  # 如果show_three为TRUE，添加自定义线条（这里以three为斜率画红色直线为例，你可以按需修改）
  if(show_three){
    abline(a = 0, b = x$three, col = "red", lwd = 2)
  }
}

现在你可以像预期那样调用plot函数了：

# 生成自定义对象
dat <- f(3, 2)

# 默认绘图：和plot(dat$one, dat$two)效果一致
plot(dat)

# 使用新增参数：添加红色线条
plot(dat, show_three = TRUE)

# 同时传递plot原生参数，比如修改坐标轴标签、点样式
plot(dat, xlab = "One (x²)", ylab = "Two (y²)", pch = 16, col = "blue", show_three = TRUE)

• 你可以根据需求修改绘图逻辑：比如默认绘制多个向量、调整线条样式、添加图例等，都可以在plot.my_custom_list里实现。
• S3方法是R中扩展基础函数（比如plot、print）最常用的方式，很多R包（包括你提到的survival）都是用这种方式实现自定义对象的可视化。

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