rescale.py

def rescale(p,rho = 0.001):
    
    rhoAbs = abs(rho)
    
    if abs(rho) > 10.001:
        print('Error in rho ')
        sys.exit(1)
    
    out = np.zeros_like(p)
    
    
    out = p**(1. + rhoAbs)
    out = out/out.sum()
    
    if rho < 0.0:
        out = p - (out - p)
    
    
    #        out = p + (p-out)
    # If rho is -ve then flip super linear scaling
    #    out = p - (out - p)
        
        
    return out