假设有如下Person类的对象实例列表:
class Person:
def __init__(self, name, age):
self.name = name
self.age = age
p1 = Person("Tom", 20)
p2 = Person("Jerry", 20)
p3 = Person("Lucy", 19)
p4 = Person("Jack", 21)
p5 = Person("Tom", 22)
persons = [p1, p2, p3, p4, p5]
现在我们希望按照每个Person实例对象的name属性进行合并,即将同名的Person对象合并。
我们可以定义一个merge_by_attribute函数来实现:
def merge_by_attribute(obj_list, attr_name):
merged_dict = {}
for obj in obj_list:
if getattr(obj, attr_name) not in merged_dict:
merged_dict[getattr(obj, attr_name)] = [obj]
else:
merged_dict[getattr(obj, attr_name)].append(obj)
merged_list = []
for key in merged_dict:
merged_obj = merged_dict[key][0]
if len(merged_dict[key]) > 1:
for obj in merged_dict[key][1:]:
merged_obj = merge_objects(merged_obj, obj)
merged_list.append(merged_obj)
return merged_list
其中,merge_objects函数用于合并具有相同属性的两个对象,我们需要自行定义。
例如,对于上述Person类的对象实例列表,我们可以通过调用merge_by_attribute(persons, "name")来对同名的Person对象进行合并:
def merge_objects(obj1, obj2):
merged_obj = Person(obj1.name, max(obj1.age, obj2.age))
return merged_obj
merged_persons = merge_by_attribute(persons, "name")
for p in merged_persons:
print(p.name, p.age)
输出结果为:
Tom 22
Jerry 20
Lucy 19
Jack 21
其中,原列表中的两个Tom对象被合并为了一个Tom对象,且其年龄取值为20