Получение конкретного набора элементов в Python

Question

По существу, это то, что я пытаюсь сделать:

У меня есть набор, который я добавить объекты. Эти объекты имеют свой собственный метод равенства, и в наборе никогда не должно быть элемента, равного другому элементу в наборе. Однако при попытке вставить элемент, если он равен другому элементу, я хотел бы записать объединенную версию этих двух элементов. То есть, объекты имеют поле «aux», которое не рассматривается в его методе равенства. Когда я закончил добавлять вещи, я хотел бы, чтобы поле «aux» элемента содержало комбинацию всех полей «aux» равных элементов, которые я пытался добавить.

Мое мышление было, хорошо, перед добавлением элемента в набор, проверьте, уже ли он в наборе. Если это так, вытащите его из набора, объедините два элемента, а затем верните их. Однако метод удаления в наборах Python ничего не возвращает, а метод pop возвращает произвольный элемент.

Может ли я делать то, что я пытаюсь сделать с наборами в Python, или я ложное дерево (что является права дерева?)

Answer 1

Похоже, вы хотите defaultdict

from collections import defaultdict 
D = defaultdict(list) 
D[somekey].append(auxfield) 

Edit: Чтобы использовать функцию слияния, вы можете объединить код люди, приведенные в комментариях

D = {} 
for something in yourthings: 
    if something.key in D: 
     D[something.key] = something.auxfield 
    else: 
     D[something.key] = merge(D[something.key], something.auxfield)