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]]