Python OrderedSet с методом .index () - PullRequest
Новая
       17

Python OrderedSet с методом .index ()

3 голосов
/ 03 ноября 2011

Кто-нибудь знает о быстрой реализации OrderedSet для python, которая:

  • запоминает порядок вставки
  • имеет метод index () (как предложение из одного списка)

Во всех обнаруженных мной реализациях отсутствует метод .index ().

1 Ответ

6 голосов
/ 03 ноября 2011

Вы всегда можете добавить его в подкласс.Вот базовая реализация для 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 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())

Вы не можете создать подкласс collections.MutableSet, потому что вы не можете поддерживать удаление элементовиз набора и держите индексы правильными.

...