在Python中检索特定的set元素

本质上,这就是我想要做的:

我有一个要添加对象的集合.这些对象具有自己的相等方法,并且集合中绝对不应包含与集合中另一个元素相同的元素.但是,当尝试插入一个元素时,如果它等于另一个元素,我想记录两个元素的合并版本.也就是说,对象具有一个“ aux”字段,该字段在其相等方法中未考虑.完成添加操作后,我希望元素的“ a​​ux”字段包含我尝试添加的相等元素的所有“ 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)
上一篇:leetcode力扣78. 子集


下一篇:Python-生成符合条件的大集合组合的最有效方法?