python中set集合判断是否重复

发布时间:2020-07-30编辑:脚本学堂
本文介绍下,在python编程中对set集合判断是否重复的方法,有需要的朋友参考下。

以下的代码中,两个值相同的Item对象,添加到set中被认为是两个对象。
 
代码:
 

复制代码 代码示例:
class Item(object):  
    def __init__(self, foo, bar): 
        self.foo = foo 
        self.bar = bar 
     
    def __repr__(self): 
        return "Item(%s, %s)" % (self.foo, self.bar) 
     
print set([Item('1', '2'), Item('1', '2')]) 
 
# 输出: set([Item(1, 2), Item(1, 2)]) 
 

数据结构中HASH表的设计思路:
被插入元素根据hash值决定插入那个桶(通常一个数组)中。
新插入元素根据的equals方法依次比较桶内元素,以确定是否已经存在。
 
python中,如果自定义class,则需要添加__hash__和__eq__两个方法,前者用于决定当前元素在哪个桶,后者决定当前元素是否在桶中已经存在。
 
修改后的代码:
 

复制代码 代码示例:
class Item(object): 
    def __init__(self, foo, bar): 
        self.foo = foo 
        self.bar = bar 
     
    def __repr__(self): 
        return "Item(%s, %s)" % (self.foo, self.bar) 
     
    def __eq__(self, other): 
        if isinstance(other, Item): 
            return ((self.foo == other.foo) and (self.bar == other.bar)) 
        else: 
            return False 
     
    def __hash__(self): 
        return hash(self.foo + " " + self.bar) 
 
print set([Item('1', '2'), Item('1', '2')]) 
 
# 输出:set([Item(1, 2)]) 

备注:
在SQLAlchemy使用model过程发现,model添加到set后会做distinct动作。
原因在于:持久化类的父类Model中已实现上面两个方法,它们默认通过主键来判断元素是不是一个。