fttools.py 文件源码

def pad2d(array, factor=1, value=0):
    ''' Symmetrically pads a 2D array with a value.

    Args:
        array (`numpy.ndarray`): source array.

        factor (`number`): number of widths of source array to add to each side (L/R/U/D).

        value (`number`): value with which to pad the array.

    Returns
        `numpy.ndarray`: padded array.

    '''
    x, y = array.shape
    pad_shape = ((int(x * factor), int(x * factor)), (int(y * factor), int(y * factor)))
    return np.pad(array, pad_width=pad_shape, mode='constant', constant_values=value)