有没有人知道python的快速OrderedSet实现:
我找到的所有实现都缺少.index()方法。
发布于 2011-11-03 08:57:48
您可以将其添加到子类中。下面是您在注释中链接的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
抽象类进行子类,以获得其他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
,因为您不能支持从集合中移除元素并保持索引正确。
https://stackoverflow.com/questions/7998692
复制