Групповой список целых последовательностей
У меня есть список целых чисел...
[1,2,3,4,5,8,9,10,11,200,201,202]
Я хотел бы сгруппировать их в список списков, где каждый подсписщик содержит целые числа, последовательность которых не была нарушена. Как это...
[[1,5],[8,11],[200,202]]
У меня довольно неуклюжая работа...
lSequenceOfNum = [1,2,3,4,5,8,9,10,11,200,201,202]
lGrouped = []
start = 0
for x in range(0,len(lSequenceOfNum)):
if x != len(lSequenceOfNum)-1:
if(lSequenceOfNum[x+1] - lSequenceOfNum[x]) > 1:
lGrouped.append([lSequenceOfNum[start],lSequenceOfNum[x]])
start = x+1
else:
lGrouped.append([lSequenceOfNum[start],lSequenceOfNum[x]])
print lGrouped
Это лучшее, что я мог сделать. Есть ли более "питонический" способ сделать это? Спасибо..
Ответы
Ответ 1
Предполагая, что список всегда будет в порядке возрастания:
from itertools import groupby, count
numberlist = [1,2,3,4,5,8,9,10,11,200,201,202]
def as_range(g):
l = list(g)
return l[0], l[-1]
print [as_range(g) for _, g in groupby(numberlist, key=lambda n, c=count(): n-next(c))]
Ответ 2
Я понял, что я немного усложнил это, гораздо проще просто пересчитать вручную, чем использовать слегка запутанный генератор:
def ranges(seq):
start, end = seq[0], seq[0]
count = start
for item in seq:
if not count == item:
yield start, end
start, end = item, item
count = item
end = item
count += 1
yield start, end
print(list(ranges([1,2,3,4,5,8,9,10,11,200,201,202])))
Производство:
[(1, 5), (8, 11), (200, 202)]
Этот метод довольно быстрый:
Этот метод (и старый, они выполняют почти то же самое):
python -m timeit -s "from test import ranges" "ranges([1,2,3,4,5,8,9,10,11,200,201,202])"
1000000 loops, best of 3: 0.47 usec per loop
Метод Джеффа Меркадо:
python -m timeit -s "from test import as_range; from itertools import groupby, count" "[as_range(g) for _, g in groupby([1,2,3,4,5,8,9,10,11,200,201,202], key=lambda n, c=count(): n-next(c))]"
100000 loops, best of 3: 11.1 usec per loop
Это более чем в 20 раз быстрее - хотя, естественно, если скорость не имеет значения, это не является реальной проблемой.
Мое старое решение с использованием генераторов:
import itertools
def resetable_counter(start):
while True:
for i in itertools.count(start):
reset = yield i
if reset:
start = reset
break
def ranges(seq):
start, end = seq[0], seq[0]
counter = resetable_counter(start)
for count, item in zip(counter, seq): #In 2.x: itertools.izip(counter, seq)
if not count == item:
yield start, end
start, end = item, item
counter.send(item)
end = item
yield start, end
print(list(ranges([1,2,3,4,5,8,9,10,11,200,201,202])))
Производство:
[(1, 5), (8, 11), (200, 202)]
Ответ 3
Вы можете сделать это эффективно в три этапа
дано
list1=[1,2,3,4,5,8,9,10,11,200,201,202]
Вычислить разрывы
[1,2,3,4,5,8,9,10,11 ,200,201,202]
- [1,2,3,4,5,8,9 ,10 ,11 ,200,201,202]
----------------------------------------
[1,1,1,1,3,1,1 ,1 ,189,1 ,1]
(index) 1 2 3 4 5 6 7 8 9 10 11
* *
rng = [i+1 for i,e in enumerate((x-y for x,y in zip(list1[1:],list1))) if e!=1]
>>> rng
[5, 9]
Добавьте границы
rng = [0] + rng + [len(list1)]
>>> rng
[0, 5, 9,12]
теперь вычислить фактические диапазоны непрерывности
[(list1[i],list1[j-1]) for i,j in zip(list2,list2[1:])]
[(1, 5), (8, 11), (200, 202)]
LB [0, 5, 9, 12]
UB [0, 5, 9, 12]
-----------------------
indexes (LB,UB-1) (0,4) (5,8) (9,11)
Ответ 4
Вопрос довольно старый, но я думал, что поделюсь своим решением в любом случае
Предполагая import numpy as np
a = [1,2,3,4,5,8,9,10,11,200,201,202]
np.split(a, array(np.add(np.where(diff(a)>1),1)).tolist()[0])
Ответ 5
псевдокод (с ошибками за один раз):
jumps = new array;
for idx from 0 to len(array)
if array[idx] != array[idx+1] then jumps.push(idx);
Я думаю, что это на самом деле случай, когда имеет смысл работать с индексами (как в C, до улучшения java/python/perl/и т.д.) вместо объектов в массиве.
Ответ 6
Здесь версия, которую нужно легко прочитать:
def close_range(el, it):
while True:
el1 = next(it, None)
if el1 != el + 1:
return el, el1
el = el1
def compress_ranges(seq):
iterator = iter(seq)
left = next(iterator, None)
while left is not None:
right, left1 = close_range(left, iterator)
yield (left, right)
left = left1
list(compress_ranges([1, 2, 3, 4, 5, 8, 9, 10, 11, 200, 201, 202]))
Ответ 7
Похожие вопросы:
Python - найти инкрементные нумерованные последовательности со списком понятий
Pythonic способ преобразования списка целых чисел в строку разделенных запятыми диапазонов
input = [1, 2, 3, 4, 8, 10, 11, 12, 17]
i, ii, result = iter(input), iter(input[1:]), [[input[0]]]
for x, y in zip(i,ii):
if y-x != 1:
result.append([y])
else:
result[-1].append(y)
>>> result
[[1, 2, 3, 4], [8], [10, 11, 12], [17]]
>>> print ", ".join("-".join(map(str,(g[0],g[-1])[:len(g)])) for g in result)
1-4, 8, 10-12, 17
>>> [(g[0],g[-1])[:len(g)] for g in result]
[(1, 4), (8,), (10, 12), (17,)]