pad-regular.py

def pad_regular(array, pad_width, highlevel=True, behavior=None):
    # If not regular, could convert to indexedarray and update indices
    def apply(array):
        array = ak.to_regular(array, highlevel=False)
        local_ix = np.pad(np.arange(array.size), pad_width, mode="edge")
        ix = np.lib.stride_tricks.as_strided(
            local_ix, shape=(len(array), len(local_ix)), strides=(0,) + local_ix.strides
        )
        return array[ak.from_regular(ix)]

    return apply_at_axis(array, apply, axis=-1, highlevel=highlevel, behavior=behavior)