R/rbind_dfs.R

In ufs: A Collection of Utilities

#' Simple alternative for rbind.fill or bind_rows
#'
#' @param x One dataframe
#' @param y Another dataframe
#' @param clearRowNames Whether to clear row names (to avoid duplication)
#'
#' @return The merged dataframe
#' @export
#'
#' @examples rbind_dfs(Orange, mtcars);
rbind_dfs <- function(x,
                      y,
                      clearRowNames = TRUE) {

  ### Check what we got

  x_notnull <- !is.null(x);
  y_notnull <- !is.null(y);

  if (x_notnull) x_df <- is.data.frame(x) else x_df <- FALSE;
  if (y_notnull) y_df <- is.data.frame(y) else y_df <- FALSE;

  x_valid <- x_notnull && x_df;
  y_valid <- y_notnull && y_df;

  if (x_valid) x_hasrows <- (nrow(x) > 0) else x_hasrows <- FALSE;
  if (y_valid) y_hasrows <- (nrow(y) > 0) else y_hasrows <- FALSE;

  if (x_hasrows && y_hasrows) {

    ### Do what we came here to do

    xNames <- names(x);
    yNames <- names(y);

    resNames <- c(xNames,
                  setdiff(yNames, xNames));

    inX_not_inY <- setdiff(xNames, yNames);
    inY_not_inX <- setdiff(yNames, xNames);

    if (length(inY_not_inX) > 0) {
      xComplement <-
        as.data.frame(lapply(inY_not_inX,
                             function(cols) {
                               return(rep(NA, nrow(x)));
                             }));
      names(xComplement) <- inY_not_inX;
      fullX <- cbind(x, xComplement)[, resNames];
    } else {
      fullX <- x[, resNames];
    }

    if (length(inX_not_inY) > 0) {
      yComplement <-
        as.data.frame(lapply(inX_not_inY,
                             function(cols) {
                               return(rep(NA, nrow(y)));
                             }));
      names(yComplement) <- inX_not_inY;
      fullY <- cbind(y, yComplement)[, resNames];
    } else {
      fullY <- y[, resNames];
    }

    res <-
      rbind(fullX, fullY);

    if (clearRowNames) {
      row.names(res) <- NULL;
    } else {
      row.names(res) <-
        c(row.names(x), row.names(y));
    }

    return(as.data.frame(res));


  } else if (x_hasrows) {

    return(x);

  } else if (y_hasrows) {

    return(y);

  } else if (x_df) {

    return(x);

  } else if (y_df) {

    return(y);

  } else {

    return(NULL);

  }

}