检测列表中的连续整数

Question

我有一个列表，里面包含了一些数据，如下所示：

[1, 2, 3, 4, 7, 8, 10, 11, 12, 13, 14]

我想把连续的整数范围打印出来：

1-4, 7-8, 10-14

有没有什么现成的、快速的、有效的方法可以做到这一点呢？

Answer 1

这段代码会按照你指定的方式输出：

>>> nums = [1, 2, 3, 4, 7, 8, 10, 11, 12, 13, 14]
>>> ranges = sum((list(t) for t in zip(nums, nums[1:]) if t[0]+1 != t[1]), [])
>>> iranges = iter(nums[0:1] + ranges + nums[-1:])
>>> print ', '.join([str(n) + '-' + str(next(iranges)) for n in iranges])
1-4, 7-8, 10-14

如果列表中有任何单个数字范围，它们会显示为 n-n 的格式：

>>> nums = [1, 2, 3, 4, 5, 7, 8, 9, 12, 15, 16, 17, 18]
>>> ranges = sum((list(t) for t in zip(nums, nums[1:]) if t[0]+1 != t[1]), [])
>>> iranges = iter(nums[0:1] + ranges + nums[-1:])
>>> print ', '.join([str(n) + '-' + str(next(iranges)) for n in iranges])
1-5, 7-9, 12-12, 15-18

Answer 2

这是一个简短的解决方案，不需要额外的导入。它可以接受任何可迭代的对象，能够对未排序的输入进行排序，并且去掉重复的项目：

def ranges(nums):
    nums = sorted(set(nums))
    gaps = [[s, e] for s, e in zip(nums, nums[1:]) if s+1 < e]
    edges = iter(nums[:1] + sum(gaps, []) + nums[-1:])
    return list(zip(edges, edges))

示例：

>>> ranges([2, 3, 4, 7, 8, 9, 15])
[(2, 4), (7, 9), (15, 15)]

>>> ranges([-1, 0, 1, 2, 3, 12, 13, 15, 100])
[(-1, 3), (12, 13), (15, 15), (100, 100)]

>>> ranges(range(100))
[(0, 99)]

>>> ranges([0])
[(0, 0)]

>>> ranges([])
[]

这个方法和@dansalmo的解决方案是一样的，我觉得这个方法很棒，虽然有点难以阅读和应用（因为它不是以函数的形式给出的）。

值得注意的是，它可以很容易地修改成输出“传统”的开放区间 [start, end)，比如通过改变返回语句来实现：

    return [(s, e+1) for s, e in zip(edges, edges)]

Answer 3

来自文档：

>>> from itertools import groupby
>>> from operator import itemgetter
>>> data = [ 1, 4,5,6, 10, 15,16,17,18, 22, 25,26,27,28]
>>> for k, g in groupby(enumerate(data), lambda (i, x): i-x):
...     print map(itemgetter(1), g)
...
[1]
[4, 5, 6]
[10]
[15, 16, 17, 18]
[22]
[25, 26, 27, 28]

你可以很简单地调整这个代码，来打印出一系列的范围。