python - 如何在 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=用于按自定义函数排序的参数，但我无法让它工作：

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