python - 如何使用python从列表中仅选择连续的数字

Question

我做了研究并尝试了下面的代码。它几乎有效，但我只想要结果中的连续数字。那将是[100,101,102]。我不想要结果中的 [75]、[78]、[109]。

from operator import itemgetter
from itertools import groupby
data = [75, 78, 100, 101, 102, 109]
for k, g in groupby(enumerate(data), lambda (i,x):i-x):
    print map(itemgetter(1), g)

上面的打印输出是:
[75]

[78]

[100, 101, 102]

[109]

我该怎么做才能得到 [100, 101, 102]？

Answer 1

示例中的

g 是元素的可迭代(例如，[75] 或 [100, 101, 102])。如果您只需要连续的数字s，听起来您希望打印 g 中存在多个元素的所有 g [注意，g 实际上是一个可迭代对象，但我们可以使用 list() 快速将其转换为列表，以获取少量元素。我们只需要保存内容，因为一个元素不能从迭代器读取两次]

尝试将 print map(itemgetter(1), g) 包装在 if 语句中，例如:

x = list(g)
if len(x) > 1:
    print map(itemgetter(1), x)