Объединить смежное число в списке в python

Question

У меня есть список, содержащий случайное число int. Я хотел бы перебрать этот список, и если число и последовательное число находятся в пределах одного числового шага друг друга, я хотел бы объединить их в подсписку.

Например:

input = [1,2,4,6,7,8,10,11] 
output = [[1,2],[4],[6,7,8],[10,11]] 

В списке ввода всегда будет содержать положительные Интс отсортированных в порядке возрастания. Я пробовал код с here.

initerator = iter(inputList) 
outputList = [c + next(initerator, "") for c in initerator] 

Хотя я могу Concat каждые две записи в списке, я не могу добавить подходящий if в списке понимания.

Python версии = 3,4

Answer 1

Хороший способ (нашел "расщепление" индексы, а затем нарежьте:.

input = [1,2,4,6,7,8,10,11] 
idx = [0] + [i+1 for i,(x,y) in enumerate(zip(input,input[1:])) if x+1!=y] + [len(input)] 
[ input[u:v] for u,v in zip(idx, idx[1:]) ] 
#output: 
[[1, 2], [4], [6, 7, 8], [10, 11]] 

использованием enumerate() и zip()

Answer 2

Если вы не должны иметь один вкладыш , вы можете использовать простую функцию генератора, объединяя элементы до тех пор, пока не нажмете непоследовательный элемент:

def consec(lst): 
    it = iter(lst) 
    prev = next(it) 
    tmp = [prev] 
    for ele in it: 
     if prev + 1 != ele: 
      yield tmp 
      tmp = [ele] 
     else: 
      tmp.append(ele) 
     prev = ele 
    yield tmp 

Выход:

In [2]: lst = [1, 2, 4, 6, 7, 8, 10, 11] 

In [3]: list(consec(lst)) 
Out[3]: [[1, 2], [4], [6, 7, 8], [10, 11]] 

Answer 3

Простейшая версия у меня есть без импорта:

def mergeAdjNum(l): 
    r = [[l[0]]] 
    for e in l[1:]: 
     if r[-1][-1] == e - 1: 
      r[-1].append(e) 
     else: 
      r.append([e]) 
    return r 

Около 33% быстрее, чем лайнеры.

Это один обрабатывает префикс символов группировки, упомянутые в комментариях:

def groupPrefStr(l): 
    pattern = re.compile(r'([a-z]+)([0-9]+)') 
    r = [[l[0]]] 
    pp, vv = re.match(pattern, l[0]).groups() 
    vv = int(vv) 
    for e in l[1:]: 
     p,v = re.match(pattern, e).groups() 
     v = int(v) 
     if p == pp and v == vv + 1: 
      r[-1].append(e) 
     else: 
      pp, vv = p, v 
      r.append([e]) 
    return r 

Это путь медленнее, чем число только один. Знать точный формат префикса (только один символ?) Может помочь избежать использования модуля re и ускорить его.