scripts to stack two numpy arrays of possibly unequal dimensions with options for filling and dealing with truncating

hstack_uneven.py

import numpy as np

def hstack_uneven(a,b,mxn=max,align='top',fill=0):
    heighta, widtha = a.shape[0], a.shape[1]
    heightb, widthb = b.shape[0], b.shape[1]
    
    outShape = np.array(a.shape)
    outShape[0] = mxn(heighta,heightb)
    outShape[1] = widtha+widthb
    out = np.zeros(outShape,dtype=a.dtype)+fill
    if align=='top':
        if mxn==max:
            out[:heighta,:widtha,...] = a
            out[:heightb,widtha:,...] = b
        if mxn==min:
            out[:,:widtha,...] = a[:min(heighta,heightb),...]
            out[:,widtha:,...] = b[:min(heighta,heightb),...]
    elif align=='bottom':
        if mxn==max:
            out[-heighta:,:widtha,...] = a
            out[-heightb:,widtha:,...] = b
        if mxn==min:
            out[:,:widtha,...] = a[-min(heighta,heightb):,...]
            out[:,widtha:,...] = b[-min(heighta,heightb):,...]
    else:
        raise TypeError("Expected 'top' or 'bottom' for align argument, got '%s'" % (align,))
    return out

vstack_uneven

import numpy as np

def vstack_uneven(a,b,mxn=max,align='left',fill=0):
    heighta, widtha = a.shape[0], a.shape[1]
    heightb, widthb = b.shape[0], b.shape[1]
    
    outShape = np.array(a.shape)
    outShape[0] = heighta+heightb
    outShape[1] = mxn(widtha,widthb)
    out = np.zeros(outShape,dtype=a.dtype)+fill
    if align=='left':
        if mxn==max:
            out[:heighta,:widtha,...] = a
            out[heighta:,:widthb,...] = b
        if mxn==min:
            out[:heighta,...] = a[:,:min(widtha,widthb),...]
            out[heighta:,...] = b[:,:min(widtha,widthb),...]
    elif align=='right':
        if mxn==max:
            out[:heighta,-widtha:,...] = a
            out[heighta:,-widthb:,...] = b
        if mxn==min:
            out[:heighta,...] = a[:,-min(widtha,widthb):,...]
            out[heighta:,...] = b[:,-min(widtha,widthb):,...]
    else:
        raise TypeError("Expected 'left' or 'right' for align argument, got '%s'" % (align,))
    return out