如何在 Python sortedcontainers 中按键正确使用 SortedSets

Question

SortedListWithKey可以使用lambda函数对列表进行排序：

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

此排序集将按元素对中的第一个索引进行排序。元组的好处是它们是可哈希的，缺点是它们是不可变的。

请考虑使用已排序的容器。SortedDict 改为：

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

排序的字典将按排序顺序保持键，并允许您将值更新为所需的任何值。