问如何在python中将符合一定条件的列表合并并创建为嵌套列表？

EN

c = zip(a, b)

c = {k: [bi for ai, bi in g] for k, g in groupby(c, lambda i: i[0])}

c = [v for k, v in sorted(c.items())]

def split_max_gap(l, max_gap=2):
    acc = [l[0]]
    for x, y in zip(l, l[1:]):
        if abs(x - y) > max_gap:
            yield acc
            acc = [y]
            continue
        acc.append(y)
    if acc:
        yield acc

c = map(split_max_gap, c)

c = list(chain.from_iterable(c))

[[1, 3], [2, 3], [6, 7]]

a = [0, 0, 2, 2, 2, 2]
b = [1, 3, 2, 3, 6, 7]
c = []

for s in set(a):
    i = a.index(s)
    count = a.count(s)
    pom = []

    for j in range(i, i + count):
        if not pom:
            pom.append(b[j])
        elif abs(pom[-1] - b[j]) < 3:
                pom.append(b[j])
                if j + 1 == count:
                    pom.insert(0, s)
                    c.append(pom)
                    pom = []
                else:
                    pom.insert(0, s)
                    c.append(pom)
                    pom = [b[j]]

print(c)