python连续数字分为一组_python – 在列表中标识不同连续数字的组

您可以创建一个迭代器来帮助分组,并尝试从下一组中提取下一个元素,该组将是上一组的结尾：

def ranges(lst):

it = iter(lst)

next(it) # move to second element for comparison

grps = groupby(lst, key=lambda x: (x - next(it, -float("inf"))))

for k, v in grps:

i = next(v)

try:

step = next(v) - i # catches single element v or gives us a step

nxt = list(next(grps)[1])

yield xrange(i, nxt.pop(0), step)

# outliers or another group

if nxt:

yield nxt[0] if len(nxt) == 1 else xrange(nxt[0], next(next(grps)[1]), nxt[1] - nxt[0])

except StopIteration:

yield i # no seq

给你：

In [2]: l1 = [2, 3, 4, 5, 8, 10, 12, 14, 13, 14, 15, 16, 17, 20, 21]

In [3]: l2 = [2, 4, 6, 8, 12, 13, 14, 15, 16, 17, 20]

In [4]: l3 = [13, 14, 15, 16, 17, 18]

In [5]: s1 = [i + 10 for i in xra