Python-найдите одинаковые значения в списке и сгруппируйте новый список

Я застрял, выясняя это, и интересно, может ли кто-нибудь указать мне в правильном направлении...

из этого списка:

N = [1,2,2,3,3,3,4,4,4,4,5,5,5,5,5]

Я пытаюсь создать:

L = [[1],[2,2],[3,3,3],[4,4,4,4],[5,5,5,5,5]]

любое значение, которое найдено одинаковым, группируется в собственный подлист. Вот моя попытка до сих пор, я думаю, что я должен использовать while петли?

global n

n = [1,2,2,3,3,3,4,4,4,4,5,5,5,5,5] #Sorted list
l = [] #Empty list to append values to

def compare(val):
   """ This function receives index values
   from the n list (n[0] etc) """

   global valin
   valin = val

   global count
   count = 0

    for i in xrange(len(n)):
        if valin == n[count]: # If the input value i.e. n[x] == n[iteration]
            temp = valin, n[count]
             l.append(temp) #append the values to a new list
             count +=1
        else:
          count +=1


for x in xrange (len(n)):
    compare(n[x]) #pass the n[x] to compare function

5 ответов


кто-то упоминает на N=[1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 5, 1] он будет получать [[1], [2, 2], [3, 3, 3], [4, 4, 4, 4], [5, 5, 5, 5, 5], [1]]

другими словами, когда номера списка не в порядке или это список беспорядка, он недоступен.

поэтому у меня есть лучший ответ, чтобы решить эту проблему.

from collections import Counter

N = [1,2,2,3,3,3,4,4,4,4,5,5,5,5,5]
C = Counter(N)

print [ [k,]*v for k,v in C.items()]

сохраняйте спокойствие и используйте itertools.groupby:

from itertools import groupby

N = [1,2,2,3,3,3,4,4,4,4,5,5,5,5,5]

print([list(j) for i, j in groupby(N)])

выход:

[[1], [2, 2], [3, 3, 3], [4, 4, 4, 4], [5, 5, 5, 5, 5]]

Примечание: запретить использование глобальной переменной, когда вы не нужно to.


можно использовать itertools.groupby вместе с понимание

>>> l =  [1,2,2,3,3,3,4,4,4,4,5,5,5,5,5]
>>> [list(v) for k,v in itertools.groupby(l)]
[[1], [2, 2], [3, 3, 3], [4, 4, 4, 4], [5, 5, 5, 5, 5]]

это может быть присвоено переменной L а в

L = [list(v) for k,v in itertools.groupby(l)]

ты все усложняешь.

то, что вы хотите сделать: для каждого значения, если это то же самое, что и последнее значение, просто добавьте его в список последних значений; в противном случае создайте новый список. Вы можете перевести напрямую на Python:

new_list = []
for value in old_list:
    if new_list and new_list[-1][0] == value:
        new_list[-1].append(value)
    else:
        new_list.append([value])

есть еще более простые способы сделать это, если вы хотите получить немного более абстрактные, например, используя функции группировки в itertools. Но это должно быть легко понять.


если вы действительно нужно сделать это с помощью while loop, вы можете перевести любой for петля в while цикл вроде этого:

for value in iterable:
    do_stuff(value)

iterator = iter(iterable)
while True:
    try:
        value = next(iterator)
    except StopIteration:
        break
    do_stuff(value)

или, если вы знаете, что iterable-это последовательность, вы можете использовать немного более простой while петли:

index = 0
while index < len(sequence):
    value = sequence[index]
    do_stuff(value)
    index += 1

но как это делает ваш код менее читаемым, менее подходящие для Python, более сложный и менее эффективный, легче ошибиться и т. д.


другое немного другое решение, которое не зависит от itertools:

#!/usr/bin/env python

def group(items):
    """
    groups a sorted list of integers into sublists based on the integer key
    """
    if len(items) == 0:
        return []

    grouped_items = []
    prev_item, rest_items = items[0], items[1:]

    subgroup = [prev_item]
    for item in rest_items:
        if item != prev_item:
            grouped_items.append(subgroup)
            subgroup = []
        subgroup.append(item)
        prev_item = item

    grouped_items.append(subgroup)
    return grouped_items

print group([1,2,2,3,3,3,4,4,4,4,5,5,5,5,5])
# [[1], [2, 2], [3, 3, 3], [4, 4, 4, 4], [5, 5, 5, 5, 5]]