python - 如何在Python sortedContainers中按键正确使用SortedSets

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（）仅具有唯一值，documentation说它也接受key =参数，以通过自定义函数进行排序，但是我无法使其正常工作：

from sortedcontainers import SortedSet

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

将抛出以下异常：

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

有没有办法做到这一点？

最佳答案

排序集要求元素是可哈希的。您的元素是不支持哈希的列表。将元素更改为元组，它将起作用：

>>> 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将使键保持排序的顺序，并让您将值更新为所需的任何值。