Group Neighbor Elements

GroupNeighbor.py

def GroupNeighbor(l):
    """
    Eg: Input <= np.array
    l = np.array([58,59,
                  87,98,
                  101,102,103,104,
                  111,122,
                  133,134,135,136])
    GroupNeighbor(l) = [[58, 59], 
                        [101, 102, 103, 104], 
                        [133, 134, 135, 136]]
    """
    NearIndex = lambda ll: np.array([value -ll[index-1]  for index, value in enumerate(ll) if index != 0])
    ReshapeIndex = lambda ll: np.append(0, ll +1)
    AppendZero = lambda ll: np.append(ll,0)
    AppendOne = lambda ll: np.append(ll,ll[-1]+1)
    u = np.argwhere(NearIndex(l) == 1)[:,0]
    t = ReshapeIndex(np.argwhere(NearIndex(AppendZero(u)) != 1)[:,0])
    return [ l[AppendOne(index)] for index in [u[t[i]:t[i+1]] for i in range(t.size-1)] ]