【Python • 集合】如何使复杂类型对象支持集合操作

简介: 【Python • 集合】如何使复杂类型对象支持集合操作

项目场景:

有两个集合,每次运算时,都需要对两个集合取差集找出两个集合的差异,或者取并集用来去重,其中的元素都是复杂类型的对象。


问题描述

众所周知,集合是无序的,且没有重复的。在本例中,定义了一个class,然后创建对象,每次往定义的结合中加元素时,同样的元素,加到集合却可以无限添加,通过调试发现,每次添加的元素Hash值是不同的。

首先定义一个Person类,用于测试,

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def get_name(self):
        return self.name
    def get_age(self):
        return self.age

在main.py中新建一个集合,向集合中添加Person类型的对象,在person_list中是所有将要添加的对象,其中有一个是重复的,是为了验证是否可以去重专门添加的,

from Model.Person import Person
def main():
    person_set = set()
    person_list = [
        Person("张三", 24),
        Person("李四", 15),
        Person("王五", 28),
        Person("张麻子", 30),
        Person("二营长", 19),
        Person("张三", 24)
    ]
    for p in person_list:
        person_set.add(p)
    print("Person set")
    for p in person_set:
        print(p)
if __name__ == '__main__':
    main()

运行后,结果如下,可以看到,在集合person_set中添加了6个对象,每个集合的hash值都是不一样的,

此时,再加一个test_person_set,用来

test_person_list = [
        Person("李四", 15),
        Person("王五", 28),
        Person("张麻子", 30),
    ]
    test_person_set = set()
    for person in test_person_list:
        test_person_set.add(person)

person_set做差,

result_set = person_set - test_person_set
    print("Result Person set")
    for p in result_set:
        print(p)

此时,输出运行结果,还是person_set中的6个元素,


原因分析:

在Python的对象中,有__str__方法,用来将对象转换成字符串,相当于java的toString()方法,__eq__方法,提供两个对象的比较运算,当使用==运算符连接两个对象时触发,而最关键的是__hash__方法,获取对象的hash值。

在Python的集合操作中,判断两个元素是否相等,应该是用对象的hash值来判断是否是同一个对象(只是猜测,并未找到实质证据,也未找到Python的源码),并且来使用eq来判断是否两个元素相等。


解决方案:

基于以上猜测,我在Person类中加入了eqhash方法,两者任意一个缺失都不能解决这个问题,Person类的对象最终如下,

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def get_name(self):
        return self.name
    def get_age(self):
        return self.age
    def __str__(self):
        return self.get_name() + str(self.get_age())
    def __eq__(self, other):
        return self.name == other.name and self.age == other.age
    def __hash__(self):
        return hash(self.name+str(self.get_age()))

此时再次运行main.py,运行结果就正确了,

完整代码:

目录结构

main.py

from Model.Person import Person
def main():
    person_set = set()
    person_list = [
        Person("张三", 24),
        Person("李四", 15),
        Person("王五", 28),
        Person("张麻子", 30),
        Person("二营长", 19),
        Person("张三", 30)
    ]
    test_person_list = [
        Person("李四", 15),
        Person("王五", 28),
        Person("张麻子", 30),
    ]
    test_person_set = set()
    for person in test_person_list:
        test_person_set.add(person)
    for p in person_list:
        person_set.add(p)
    result_set = person_set - test_person_set
    print("Result Person set")
    for p in result_set:
        print(p)
if __name__ == '__main__':
    main()

person.py

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def get_name(self):
        return self.name
    def get_age(self):
        return self.age
    def __str__(self):
        return self.get_name() + str(self.get_age())
    def __eq__(self, other):
        return self.name == other.name and self.age == other.age
    def __hash__(self):
        return hash(self.name+str(self.get_age()))
目录
相关文章
|
2月前
|
Python
python对象模型
这篇文章介绍了Python中的对象模型,包括各种内置对象类型如数字、字符串、列表、字典等,以及如何使用`type()`函数来查看变量的数据类型。
|
4天前
|
Python
以下是一些常用的图表类型及其Python代码示例,使用Matplotlib和Seaborn库。
以下是一些常用的图表类型及其Python代码示例,使用Matplotlib和Seaborn库。
|
2月前
|
Python
探索Python中的魔法方法:打造你自己的自定义对象
【8月更文挑战第29天】在Python的世界里,魔法方法如同神秘的咒语,它们赋予了对象超常的能力。本文将带你一探究竟,学习如何通过魔法方法来定制你的对象行为,让你的代码更具魔力。
39 5
|
2天前
|
存储 API 索引
Python 的集合是怎么实现的?
Python 的集合是怎么实现的?
24 9
|
3天前
|
存储 索引 Python
Python常用数据结构——集合
Python常用数据结构——集合
17 3
|
7天前
|
关系型数据库 MySQL 数据库
Python MySQL查询返回字典类型数据的方法
通过使用 `mysql-connector-python`库并选择 `MySQLCursorDict`作为游标类型,您可以轻松地将MySQL查询结果以字典类型返回。这种方式提高了代码的可读性,使得数据操作更加直观和方便。上述步骤和示例代码展示了如何实现这一功能,希望对您的项目开发有所帮助。
25 4
|
5天前
|
测试技术 数据库 开发者
Python作为一种谦逊的编程语言:对象自省机制的探讨
Python的自省机制是该语言的一个强大特性,为开发者提供了深入了解和操作对象的能力。它增强了Python的灵活性,使得开发者可以更加精准地控制程序的行为。然而,合理利用自省能力,避免其成为代码复杂性的来源,是每个Python开发者需要考虑的问题。通过熟练运用Python提供的自省工具和技巧,可以更好地设计和实现高效、易维护的Python应用。
13 2
|
4天前
|
存储 数据处理 Python
Python中的Set集合:高效数据处理的利器
Python中的Set集合:高效数据处理的利器
12 0
|
5天前
|
Python
python推导式-列表,元组,字典,集合推导式
这篇文章介绍了Python中的推导式,包括列表推导式、元组推导式、字典推导式和集合推导式,提供了它们的基本格式和示例代码,并解释了推导式如何简化循环和条件判断的代码编写。
|
2月前
|
Python