code: def columns(x, rho, proxop):
      
  
      xnext = np.zeros_like(x)
  
      for ix in range(x.shape[1]):
          xnext[:, ix] = proxop(x[:, ix], rho)
  
      return xnext
docs: Applies a proximal operator to the columns of a matrix