如何在 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() 来获得唯一值，documentation 表示它也接受 key= 用于按自定义函数排序的参数，但我无法使其工作：

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'

sorted dict 将使键保持排序顺序，并让您将值更新为您喜欢的任何值。

如何在 Python sortedcontainers 中正确使用 SortedSets

How to correctly use SortedSets by key in Python sortedcontainers

python

sorting

list

ordereddictionary

sortedcontainers