从python中的自定义对象列表中删除重复项的有效方法

时间:2015-02-08 23:45:42

标签: python list filtering subset

我有一个自定义类的对象,其中包含各种不同类型的各种属性。我想根据其中一个属性从这些对象的列表中删除重复项。

这样的东西,但实际上得到的是对象列表而不是指定属性的列表。

filteredData = list(set([x.attribute[0] for x in objList]))

1 个答案:

答案 0 :(得分:1)

您需要在对象上实现方法哈希 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