Как правильно использовать SortedSets по ключу в Python sortedcontainers

Question

SortedListWithKey может отсортировать список с помощью лямбда-функции:

from sortedcontainers import SortedListWithKey

SortedListWithKey([[4, 'last'], [1, 'first']], key=lambda x: x[0])
# Result: SortedListWithKey([[1, 'first'], [4, 'last']], key=<function <lambda> at 0x107f5d730>)

Но скажем, мне нужно использовать set(), чтобы иметь только уникальные значения, в документации сказано, что он также принимает key = param для сортировки по пользовательской функции, но я не могу заставить его работать:

from sortedcontainers import SortedSet

SortedSet([[4, 'last'], [1, 'first']], key=lambda x: x[0])

Скину следующее исключение:

values = set(chain(*iterables))
TypeError: unhashable type: 'list'

Есть ли способ добиться этого?

Answer 1

Сортированный набор требует, чтобы элементы были хэшируемыми. Ваши элементы - это списки, которые не поддерживают хеширование. Измените элементы на кортежи, и это будет работать:

>>> from sortedcontainers import SortedSet
>>> ss = SortedSet([(4, 'last'), (1, 'first')], key=lambda value: value[0])
>>> ss
SortedSet([(1, 'first'), (4, 'last')], key=<function <lambda> at 0x10fff4848>)

Этот отсортированный набор упорядочит элементы по первому индексу в паре. Преимущество кортежей в том, что они хэшируемы, недостаток в том, что они неизменны.

Попробуйте вместо этого использовать sortedcontainers.SortedDict:

>>> sd = SortedDict({4: 'last', 1: 'first'})
>>> sd
SortedDict({1: 'first', 4: 'last'})
>>> sd[2] = 'second'
>>> sd.pop(4)
'last'

Сортированный dict сохранит ключи в отсортированном порядке и позволит вам обновлять значения в соответствии с вашими предпочтениями.