Python OrderedSet with.index（）方法

有人知道python的fast OrderedSet实现吗

• 记住插入顺序
• 有一个index（）方法（类似于列表提供的方法）

---

我找到的所有实现都缺少.index（）方法。

您始终可以将其添加到子类中。以下是您在注释中链接的

OrderedSet

的基本实现：

class IndexOrderedSet(OrderedSet):
    def index(self, elem):
        if key in self.map:
            return next(i for i, e in enumerate(self) if e == elem)
        else:
            raise KeyError("That element isn't in the set")

您提到您只需要按迭代顺序添加、

索引

。您可以使用

OrderedDict

作为存储来实现这一点。作为奖励，您可以将

collections.Set

abstract类子类化，以获得其他Set操作

frozenset

s支持：

from itertools import count, izip
from collections import OrderedDict, Set

class IndexOrderedSet(Set):
    """An OrderedFrozenSet-like object
       Allows constant time 'index'ing
       But doesn't allow you to remove elements"""
    def __init__(self, iterable = ()):
        self.num = count()
        self.dict = OrderedDict(izip(iterable, self.num))
    def add(self, elem):
        if elem not in self:
            self.dict[elem] = next(self.num)
    def index(self, elem):
        return self.dict[elem]
    def __contains__(self, elem):
        return elem in self.dict
    def __len__(self):
        return len(self.dict)
    def __iter__(self):
        return iter(self.dict)
    def __repr__(self):
        return 'IndexOrderedSet({})'.format(self.dict.keys())

---

您不能对集合.MutableSet进行子类化，因为您不支持从集合中删除元素并保持索引正确。

您尝试了什么实现？例如，这就是它的作用所在。。。需要恒定的时间。。。我只需要.add（）和.index（）@Titusz。对于OrderedSet实现，似乎不可能使用固定时间。您必须维护一个单独的索引列表，这将破坏

放弃

等的性能。@Titusz如果您真的只需要

添加

和

索引

，那么也许您可以尝试类似我添加的内容。@agf这是一个很好的答案。。。我试用了基于dict的实现，效果很好。然后我发现，我还需要按顺序遍历键；）。我可以使用self.dict=OrderedDict（）来解决这个问题。我必须检查哪个实现更快…@Titusz我更新并扩展了基于

dict

的实现。它真的没有理由不支持所有set操作

frozenset

s支持，所以我将其更改为subclass

collections.set

。我还为它提供了一个适当的

\uuuu repr\uuuu

，这样您就可以轻松地

打印它，并简化了
\uu init\uuuu
。