Python的有序集合—如何保留集合的顺序

238 阅读2分钟

至于 Python 3.7 和 CPython 3.6,正则 dict 被证实可以保留秩序,并且比OrderedDict 表现更好。要把 dictionary 用作有序集,在保持秩序的同时过滤掉重复的元素,可以模拟一个有序集。

Python 有序集

没有作为有序集的内置功能,但许多包可以帮助你使用OrderedSet。

  1. 有序集(基于Python)
  2. orderedset(基于CPython)
  3. collections-extended
  4. boltons(在iterutils.IndexedSet下,基于Python)。
  5. oset
  6. 排序的容器

在Python中使用boltons的OrderedSet

为了使用 boltonsIndexSet,你需要安装boltons 包:

python3 -m pip install boltons

现在,使用下面的代码导入该模块:

from boltons.setutils import IndexedSet

让我们使用IndexedSet并将两个列表作为参数传递:

from boltons.setutils import IndexedSet

data = IndexedSet(list(range(2)) + list(range(3, 6)))

print(data)

输出

IndexedSet([0, 1, 3, 4, 5])

现在,让我们改变列表的顺序,同时将其传递给IndexedSet

from boltons.setutils import IndexedSet

data = IndexedSet(list(range(3, 6)) + list(range(2)))

print(data)

输出

IndexedSet([3, 4, 5, 0, 1])

你可以看到,这个集合的顺序被完美地保留了下来。它保留了它们被添加到 Python 集合中的顺序。

在 Python 中使用 sortedcontainers 的 OrderedSet

Python sortedcontainers模块提供了一个 SortedSet,用于保留集合元素的顺序。使用sortedcontainers有一些好处,包括纯Python,快速实现,100%的单元测试覆盖率,数小时的压力测试。

让我们使用pip 软件包管理器来安装sortedcontainers 模块:

python3 -m pip install sortedcontainers

现在,使用下面的代码导入该模块:

from sortedcontainers import SortedSet

让我们使用**SortedSet()**方法,将两个列表作为参数传递:

from sortedcontainers import SortedSet

data = SortedSet(list(range(2)) + list(range(3, 6)))

print(data)

输出

SortedSet([0, 1, 3, 4, 5])

你可以看到,返回的集合默认是以升序排序的。

现在,让我们在传递给SortedSet的同时改变列表的顺序:

from sortedcontainers import SortedSet

data = SortedSet(list(range(3, 6)) + list(range(2)))

print(data)

输出

SortedSet([0, 1, 3, 4, 5])

你可以看到,即使我们以不同的顺序向集合添加元素,它也会默认以升序返回。

Python中的OrderedSet就这样了。