R/collapseVisregList.R

In pbreheny/visreg: Visualization of Regression Models

collapse.visregList <- function(obj, labels, ...) {
  if (missing(labels)) {
    l <- sapply(obj, function(v) v$meta$y)
    if (any(duplicated(l))) {
      labels <- paste0("Y", 1:length(obj))
    } else {
      labels <- l
    }
  }
  if (length(labels) != length(obj)) stop("labels do not match list", call.=FALSE)
  fit <- data.frame(obj[[1]]$fit, visregCollapse=labels[1])
  res <- data.frame(obj[[1]]$res, visregCollapse=labels[1])
  if (length(obj) >= 2) {
    for (i in 2:length(obj)) {
      fit <- merge(fit, data.frame(obj[[i]]$fit, visregCollapse=labels[i]), all.x=TRUE, all.y=TRUE)
      res <- merge(res, data.frame(obj[[i]]$res, visregCollapse=labels[i]), all.x=TRUE, all.y=TRUE)
    }
  }
  meta <- obj[[1]]$meta
  meta$by <- "visregCollapse"
  structure(list(fit=fit, res=res, meta=meta), class="visreg")
}