python - 从具有连续数字的列表中创建列表列表

Question

这是我的 list

a = [ 1, 2, 3, 6, 7, 9, 11, 14, 15, 16]

我想返回一个像这样的新列表

new_a = [ [1,2,3],
          [6,7],
          [9],
          [11],
          [14,15,16]
        ]

我完全不知道如何真正实现这一目标。

这是我尝试过的。请不要笑

#!/usr/env python
a = [1,2,3,5,7,9,10,11,18,12,20,21]
a.sort()
final=[]
first = a.pop(0)
temp=[]
for i in a:
        if i - first == 1:
                temp.append(first)
                temp.append(i)
                first=i
        else:
                final.append(set(temp))
                first=i
                temp=[]
print final

[set([1, 2, 3]), set([]), set([]), set([9, 10, 11, 12]), set([])]

感谢 stackoverflow，像我这样的菜鸟可以学习 :DDD #CoDeSwAgg

Answer 1

你可以通过一些聪明的 itertools 工作来做到这一点:

[[v[1] for v in vals] for _, vals in itertools.groupby(enumerate(a), key=lambda x: x[1] - x[0])]

更自然地分解它:

result = []
groups = itertools.groupby(enumerate(a), key=lambda x: x[1] - x[0])
for _, values in groups:
    result.append([v[1] for v in values])