Сортировать список namedtupled по наиболее часто встречающимся полям - PullRequest
Новая
       17

Сортировать список namedtupled по наиболее часто встречающимся полям

0 голосов
/ 25 марта 2020

Какие есть элегантные и быстрые простые способы сортировки списка namedtuple по наиболее частым элементам в списке?

Например, у нас есть этот список 

character_list = [
    Element(id=1, character='A'),
    Element(id=2, character='B'),
    Element(id=3, character='B'),
    Element(id=4, character='C'),
    Element(id=5, character='D'),
    Element(id=6, character='E'),
    Element(id=7, character='F'),
    Element(id=8, character='H'),
    Element(id=9, character='I'),
    Element(id=10, character='J'),
    Element(id=11, character='K'),
    Element(id=12, character='L'),
    Element(id=13, character='M'),
    Element(id=14, character='J'),
    Element(id=15, character='N'),
    Element(id=16, character='J')]

И отсортировано вот так? 

character_list = [
    Element(id=10, character='J'),
    Element(id=14, character='J'),
    Element(id=16, character='J'),
    Element(id=2, character='B'),
    Element(id=3, character='B'),
    Element(id=1, character='A'),
    Element(id=4, character='C'),
    Element(id=5, character='D'),
    Element(id=6, character='E'),
    Element(id=7, character='F'),
    Element(id=8, character='H'),
    Element(id=9, character='I'),
    Element(id=11, character='K'),
    Element(id=12, character='L'),
    Element(id=13, character='M'),
    Element(id=14, character='J'),
    Element(id=15, character='N')]

Попробуйте это, но, похоже, не получите результатов, которые я ищу 

sorted(character_list, key=lambda x: character_list.count(x.character))

1 Ответ

1 голос
/ 25 марта 2020

x.character отсутствует никогда в вашем списке . В любом случае, использование list.count как это крайне неэффективно . Сортировка - O (N * log N), однако, если ваша ключевая функция использует list.count, все ухудшится до O (N ** 2).

Вместо этого создайте словарь подсчетов, и используйте этот словарь, это поддержит вашу производительность O (N * log N). Итак, учитывая: 

>>> from pprint import pprint
>>> pprint(character_list)
[Element(id=1, character='A'),
 Element(id=2, character='B'),
 Element(id=3, character='B'),
 Element(id=4, character='C'),
 Element(id=5, character='D'),
 Element(id=6, character='E'),
 Element(id=7, character='F'),
 Element(id=8, character='H'),
 Element(id=9, character='I'),
 Element(id=10, character='J'),
 Element(id=11, character='K'),
 Element(id=12, character='L'),
 Element(id=13, character='M'),
 Element(id=14, character='J'),
 Element(id=15, character='N'),
 Element(id=16, character='J')]

Тогда 

>>> from collections import Counter
>>> counts = Counter(e.character for e in character_list)
>>> counts
Counter({'J': 3, 'B': 2, 'A': 1, 'C': 1, 'D': 1, 'E': 1, 'F': 1, 'H': 1, 'I': 1, 'K': 1, 'L': 1, 'M': 1, 'N': 1})

Наконец, 

>>> def keyfunc(e):
...     return counts[e.character]
...
>>> sorted_character_list = sorted(character_list, key=keyfunc, reverse=True)
>>> pprint(sorted_character_list)
[Element(id=10, character='J'),
 Element(id=14, character='J'),
 Element(id=16, character='J'),
 Element(id=2, character='B'),
 Element(id=3, character='B'),
 Element(id=1, character='A'),
 Element(id=4, character='C'),
 Element(id=5, character='D'),
 Element(id=6, character='E'),
 Element(id=7, character='F'),
 Element(id=8, character='H'),
 Element(id=9, character='I'),
 Element(id=11, character='K'),
 Element(id=12, character='L'),
 Element(id=13, character='M'),
 Element(id=15, character='N')]
...