linear_pdhg.py 文件源码

def projsplx(y):
    y1 = np.array(y, copy=True)
    m = y1.shape[1]
    bget = False
    y1[0][::-1].sort()
    tmpsum = 0
    for ii in xrange(0,m-1):
        tmpsum = tmpsum + y1[0][ii]
        tmax = (tmpsum - 1)/ii
        if tmax >= y1[0][ii+1]:
            bget = True
            break
    if not bget:
        tmax = (tmpsum + y1[0][m] -1)/m
    y1 = y1 - tmax
    y1[y1<0.0] = 0.0
    return y1