Есть ли умный способ передать ключ defaultdict default

95

У класса есть конструктор, который принимает один параметр:

class C(object):
    def __init__(self, v):
        self.v = v
        ...

Где-то в коде для значений в dict полезно знать их ключи.
Я хочу использовать defaultdict с ключом, переданным новым значениям по умолчанию:

d = defaultdict(lambda : C(here_i_wish_the_key_to_be))

Какие-либо предложения?

python dictionary defaultdict

— Бенджамин Нитлехоо
источник

128

Это вряд ли можно назвать умным, но создание подклассов - ваш друг:

class keydefaultdict(defaultdict):
    def __missing__(self, key):
        if self.default_factory is None:
            raise KeyError( key )
        else:
            ret = self[key] = self.default_factory(key)
            return ret

d = keydefaultdict(C)
d[x] # returns C(x)

— Йохен Ритцель
источник

16

Это именно то уродство, которого я пытаюсь избежать ... Даже использование простого dict и проверка наличия ключа намного чище.

— Бенджамин Нитлехоо,

1

@ Пол: и все же это ваш ответ. Уродство? Давай!

— tzot

4

Думаю, я просто возьму этот кусок кода и вставлю его в свой персонализированный модуль общих утилит, чтобы я мог использовать его, когда захочу. Не так уж и некрасиво ...

— weronika 07

24

+1 Непосредственно отвечает на вопрос ОП и не выглядит "уродливым" мне. Кроме того, хороший ответ , потому что многие из них , кажется, не понимают , что defaultdict«s __missing__()метод может быть переопределен (как это может в любой подкласс встроенного dictкласса , начиная с версии 2.5).

— Мартино 01

7

+1 Вся цель __missing__ - настроить поведение отсутствующих ключей. Подход dict.setdefault (), упомянутый @silentghost, также будет работать (с положительной стороны, setdefault () короток и уже существует; с другой стороны, он страдает от проблем с эффективностью, и никому не нравится имя "setdefault") .

— Раймонд Хеттингер

26

Нет, нет.

defaultdictРеализация не может быть сконфигурирована для передачи отсутствует keyв default_factoryвне коробки. Ваш единственный вариант - реализовать свой собственный defaultdictподкласс, как это было предложено @JochenRitzel выше.

Но это не «умно» или почти так чисто, как было бы стандартное библиотечное решение (если бы оно существовало). Таким образом, ответ на ваш лаконичный вопрос «да / нет» будет однозначно «Нет».

Жаль, что в стандартной библиотеке отсутствует столь часто используемый инструмент.

— Стюарт Берг
источник

Да, было бы лучше, если бы фабрика могла взять ключ (унарная функция, а не нулевая). Если мы хотим вернуть константу, легко отказаться от аргумента.

— YvesgereY

6

Я не думаю, что тебе defaultdictздесь вообще нужно . Почему бы просто не использовать dict.setdefaultметод?

>>> d = {}
>>> d.setdefault('p', C('p')).v
'p'

Это, конечно, создаст много экземпляров C. Если это проблема, я думаю, подойдет более простой подход:

>>> d = {}
>>> if 'e' not in d: d['e'] = C('e')

defaultdictНасколько я понимаю, это будет быстрее, чем любая другая альтернатива.

ETA относительно скорости inтеста по сравнению с использованием предложения try-except:

>>> def g():
    d = {}
    if 'a' in d:
        return d['a']


>>> timeit.timeit(g)
0.19638929363557622
>>> def f():
    d = {}
    try:
        return d['a']
    except KeyError:
        return


>>> timeit.timeit(f)
0.6167065411074759
>>> def k():
    d = {'a': 2}
    if 'a' in d:
        return d['a']


>>> timeit.timeit(k)
0.30074866358404506
>>> def p():
    d = {'a': 2}
    try:
        return d['a']
    except KeyError:
        return


>>> timeit.timeit(p)
0.28588609450770264

— Тихий призрак
источник

7

Это очень расточительно в тех случаях, когда к d обращаются много раз, и лишь изредка пропускается ключ: C (ключ), таким образом, создаст массу ненужных объектов, которые собирает сборщик мусора. Кроме того, в моем случае возникает дополнительная боль, поскольку создание новых объектов C происходит медленно.

— Бенджамин Нитлехоо,

@ Пол: верно. Я бы предложил еще более простой метод, см. Мое редактирование.

— SilentGhost

Я не уверен, что это быстрее, чем defaultdict, но я обычно так делаю (см. Мой комментарий к ответу THC4k). Я надеялся, что есть простой способ обойти тот факт, что default_factory не принимает аргументов, чтобы код был немного более элегантным.

— Бенджамин Нитлехоо,

5

@SilentGhost: Я не понимаю - как это решает проблему OP? Я думал, что OP хочет любую попытку прочитать, d[key]чтобы вернуть d[key] = C(key)if key not in d. Но ваше решение требует, чтобы он на самом деле ехал и заранее настраивал d[key]? Как он узнал, что keyему нужно?

— max

2

Поскольку setdefault чертовски уродлив, а defaultdict из коллекции ДОЛЖЕН поддерживать фабричную функцию, которая получает ключ. Какая упущенная возможность от разработчиков Python!

— jgomo3

0

Вот рабочий пример словаря, который автоматически добавляет значение. Демонстрационная задача по поиску повторяющихся файлов в / usr / include. Обратите внимание, что для настройки словаря PathDict требуется всего четыре строки:

class FullPaths:

    def __init__(self,filename):
        self.filename = filename
        self.paths = set()

    def record_path(self,path):
        self.paths.add(path)

class PathDict(dict):

    def __missing__(self, key):
        ret = self[key] = FullPaths(key)
        return ret

if __name__ == "__main__":
    pathdict = PathDict()
    for root, _, files in os.walk('/usr/include'):
        for f in files:
            path = os.path.join(root,f)
            pathdict[f].record_path(path)
    for fullpath in pathdict.values():
        if len(fullpath.paths) > 1:
            print("{} located in {}".format(fullpath.filename,','.join(fullpath.paths)))

— Джерард
источник

Есть ли умный способ передать ключ defaultdict default_factory?