在R中使用自定义函数时如何删除一部分？

Question

我有一个自定义功能，效果很好。

Gplot = function(mydata, AE){
ggplot(mydata)+
...
geom_vline(aes(xintercept = breaks[2]), color = "seagreen", linetype = "dashed"))
}
...

有时候我不需要垂直线，那么使用时如何去掉geom_vline呢？

我知道我可以在函数中删除这句话，但大多数时候我都需要它。所以我只想用一些特定的变量来删除它。

Answer 1

使用条件并且仅在 addline 参数为 TRUE 时添加垂直线：

Gplot = function(mydata, AE, addline=TRUE){
  gp <- ggplot(mydata)+
  ...
  if(addline){
    gp <- gp + geom_vline(aes(xintercept = breaks[2]))
  }
  return(gp)
}
...