本质上,这就是我想要做的:
我有一个要添加对象的集合.这些对象具有自己的相等方法,并且集合中绝对不应包含与集合中另一个元素相同的元素.但是,当尝试插入一个元素时,如果它等于另一个元素,我想记录两个元素的合并版本.也就是说,对象具有一个“ aux”字段,该字段在其相等方法中未考虑.完成添加操作后,我希望元素的“ aux”字段包含我尝试添加的相等元素的所有“ aux”字段的组合.
我的想法是,好的,在将元素添加到集合之前,请检查它是否已经在集合中.如果是这样,请将其从集合中拉出,合并两个元素,然后再放回去.但是,Python集合中的remove方法不会返回任何内容,而pop方法会返回一个任意元素.
我可以使用Python中的集合做我想做的事情,还是我吠错了树(正确的树是什么?)
解决方法:
听起来你想要一个defaultdict
from collections import defaultdict
D = defaultdict(list)
D[somekey].append(auxfield)
编辑:
要使用合并功能,可以合并人们在注释中给出的代码
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)