plotting.py 文件源码

def pixelize(data, factor=0.1, order=0, mode="nearest", add_noise=True, noise_factor=0.3):
    h, w = np.shape(data)[0:2]

    if add_noise:
        noise = noise_factor * np.random.random(np.shape(data)) - 0.5*noise_factor
        data += noise

    factor_h = h / float(int(h)/(int(1/factor)))
    factor_w = w / float(int(w)/(int(1/factor)))
    small = sci.zoom(data, factor, order=order, mode=mode)
    pixelized = sci.zoom(small, [factor_h, factor_w], order=order, mode=mode)

    return pixelized