Python中基于公共匹配属性的对象列表合并为一致列表

Question

我有一个对象列表，我想根据匹配的属性（id）和可选的类参数“压缩”成一个更小的对象列表。在

class Case:
    def __init__(self, id, formtype, age, fever=None, cough=None, gender=None):
        self.case_id = case_id
        self.form_type = formtype
        self.age = age
        self.fever = fever
        self.cough = cough
        self.gender = gender

caselist = [
    Case(id="12345", formtype="A", age=12, fever=1, gender="female"),
    Case(id="12345", formtype="B", age=12, cough=0),
    Case(id="67890", formtype="A", age=34, fever=0, gender="male"),
    Case(id="67890", formtype="B", age=34, cough=1),
    Case(id="75321", formtype="A", age=2, fever=0, gender="male")
]

我怎么才能得到这样的新名单？它应该选择formtype="B"而不是formtype="A"。在

^{pr2}$

我试着用dict压缩它，但没有成功：

compressed = [Case(id=case.id, formtype=None, age=case.age) for event in caselist if case.formtype == 'A']

Answer 1

按id分组，并将具有“B”form_类型的对象保留为具有“B”formtype的重复id的对象，否则保持原样，如果要使用“B”中未设置的任何属性，可以使用getattr和setattr迭代属性，以设置B中以前未设置的任何属性，除非事先知道A中设置了什么和/或B中设置了什么，否则不能硬编码设置什么或不设置什么：

class Case:
    def __init__(self, id, formtype, age, fever=None, cough=None, gender=None):
        self.case_id = id
        self.form_type = formtype
        self.age = age
        self.fever = fever
        self.cough = cough
        self.gender = gender

    def __iter__(self):
        for ele in ["case_id", "form_type", "age",
                    "fever", "cough", "gender"]:
            yield ele


caselist = [
    Case(id="12345", formtype="A", age=12, fever=1, gender="female"),
    Case(id="12345", formtype="B", age=12, cough=0),
    Case(id="67890", formtype="A", age=34, fever=0, gender="male"),
    Case(id="67890", formtype="B", age=34, cough=1),
    Case(id="75321", formtype="A", age=2, fever=0, gender="male")
]

d = {}

for c in caselist:
    if c.case_id not in d:
        d[c.case_id] = c
    elif d[c.case_id].form_type != "B" and c.form_type == "B":
        tmp = d[c.case_id]
        for attr in c:
            if getattr(c, attr) is None:
                setattr(c, attr, getattr(tmp, attr))
        d[c.case_id] = c

caselist[:] = d.values()
print(caselist)