带有.index()方法的Python有序集合

Question

有没有人知道一个快速的OrderedSet实现，适用于Python，并且满足以下条件：

• 能记住元素的插入顺序
• 有一个.index()方法（就像列表那样）

我找到的所有实现都没有.index()这个方法。

Answer 1

你可以在一个子类里添加这个功能。这里有一个简单的实现，针对你在评论中提到的 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")

你提到你只需要 add（添加）、index（索引）和按顺序遍历。你可以通过使用 OrderedDict 来存储这些数据，来实现这个功能。作为额外的好处，你还可以创建一个子类，继承 collections.Set 抽象类，这样就能获得其他集合操作，像 frozenset 支持的那些：

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())

你不能继承 collections.MutableSet，因为这样无法在保持索引正确的情况下，支持从集合中移除元素。