Создать Counter
из списка, замените ключи отсортированного результата и превратить его обратно в список:
from collections import Counter
from itertools import count
# Get counts of each element in the list
original_counter = Counter([687, 687, 683, 683, 677, 662])
# Get only the unique values, in descending order
values = (v for k, v in sorted(original_counter.items(), reverse=True))
# Create a new counter out of 0, 1, 2, … and the sorted, unique values
new_counter = Counter(dict(zip(count(), values)))
# Retrieve a sorted list from the new counter
new_list = sorted(new_counter.elements())
print(new_list) # [0, 0, 1, 1, 2, 3]
Это не требует первоначальный список будет отсортирован, либо. Это делает для компактной функции:
from collections import Counter
from itertools import count
def enumerate_unique(iterable):
return sorted(Counter(dict(zip(count(),
(v for k, v in sorted(Counter(iterable).items(), reverse=True)))))
.elements())
На второй взгляд, однако, простой способ не плох. Это также немного более эффективно.
def enumerate_unique(iterable):
seen = {}
counter = 0
for x in iterable:
i = seen.get(x)
if i is None:
seen[x] = counter
yield counter
counter += 1
else:
yield i
Это работает в любом списке. Поскольку у вас есть отсортированный список, хотя, есть очень хороший O (п):
def enumerate_unique(sorted_iterable):
last = None
counter = -1
for x in sorted_iterable:
if x != last:
counter += 1
yield counter
Чтобы пропустить числа, как описано выше, вы можете сделать это:
def enumerate_unique(sorted_iterable):
last = None
last_index = -1
for i, x in enumerate(sorted_iterable):
if x != last:
last_index = i
yield last_index
просто повторите этот список. –