Как добавить количество номеров к дубликатам в списке в Python? - PullRequest
4 голосов
/ 15 мая 2010

Вот список, содержащий дубликаты:

l1 = ['a', 'b', 'c', 'a', 'a', 'b']

Вот желаемый результат:

l1 = ['a', 'b', 'c', 'a_1', 'a_2', 'b_1']

Как дубликатыпереименовать, добавив номер счета?


Вот попытка достичь этой цели;Однако есть ли более Pythonic путь?

for index in range(len(l1)):
    counter = 1
    list_of_duplicates_for_item = [dup_index for dup_index, item in enumerate(l1) if item == l1[index] and l1.count(l1[index]) > 1]
    for dup_index in list_of_duplicates_for_item[1:]: 
        l1[dup_index] = l1[dup_index] + '_' + str(counter)
        counter = counter + 1

Ответы [ 4 ]

17 голосов
/ 15 мая 2010

В Python создание нового списка обычно намного проще, чем изменение существующего списка. У нас есть генераторы, чтобы сделать это эффективно. Диктовка может вести подсчет случаев.

l = ['a', 'b', 'c', 'a', 'a', 'b']

def rename_duplicates( old ):
    seen = {}
    for x in old:
        if x in seen:
            seen[x] += 1
            yield "%s_%d" % (x, seen[x])
        else:
            seen[x] = 0
            yield x

print list(rename_duplicates(l))
5 голосов
/ 15 мая 2010

Я бы сделал что-то вроде этого:

a1 = ['a', 'b', 'c', 'a', 'a', 'b']
a2 = []

d = {}

for i in a1:

    d.setdefault(i, -1)
    d[i] += 1

    if d[i] >= 1:
        a2.append('%s_%d' % (i, d[i]))
    else:
        a2.append(i)

print a2
1 голос
/ 15 мая 2010

Я думаю, что вывод, который вы запрашиваете, сам по себе грязный, и поэтому нет чистого способа его создания.

Как вы собираетесь использовать этот новый список? Будет ли словарь подсчета похож на следующую работу?

{'a':3, 'b':2, 'c':1}

Если это так, я бы порекомендовал:

from collections import defaultdict
d = defaultdict(int) # values default to 0
for key in l1:
    d[key] += 1
0 голосов
/ 15 мая 2010

Исходя из вашего комментария к @mathmike, если вашей конечной целью является создание словаря из списка с дублирующимися ключами, я бы использовал defaultdict из `Коллекции Lib.

>>> from collections import defaultdict
>>> multidict = defaultdict(list)
>>> multidict['a'].append(1)
>>> multidict['b'].append(2)
>>> multidict['a'].append(11)
>>> multidict
defaultdict(<type 'list'>, {'a': [1, 11], 'b': [2]})
...