天天看点

python获取set中某些元素_在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)

标签:iteration,combinations,set,python

来源: https://codeday.me/bug/20191123/2066079.html