我有一个自定义的对象类,其中包含各种不同类型的属性。我希望根据其中一个属性从这些对象的列表中删除重复项。
类似于以下代码,但实际上需要获取对象列表而不是指定属性的列表。
filteredData = list(set([x.attribute[0] for x in objList]))
我有一个自定义的对象类,其中包含各种不同类型的属性。我希望根据其中一个属性从这些对象的列表中删除重复项。
类似于以下代码,但实际上需要获取对象列表而不是指定属性的列表。
filteredData = list(set([x.attribute[0] for x in objList]))
您需要在对象上实现 hash 和 eq 方法。
class A:
def __init__(self, a):
self.attr1 = a
def __hash__(self):
return hash(self.attr1)
def __eq__(self, other):
return self.attr1 == other.attr1
def __repr__(self):
return str(self.attr1)
例子:
l = [A(5), A(4), A(4)]
print list(set(l))
print list(set(l))[0].__class__ # ==> __main__.A. It's a object of class