微信公众号：「Python读财」
如有问题或建议，请公众号留言
Python中的集合和数学上的集合是一个概念，基本功能包括关系测试和消除重复元素，对于集合还可以进行数学上的交、并、差运算。定义一个集合的方式，见下面的代码：
①使用set()函数
In [18]: color_set = set(['green','blue','red','yellow','blue'])
In [20]: color_set
Out[20]: {'blue', 'green', 'red', 'yellow'}
②使用{}定义，需要注意的是{}必须含有元素，空的{}定义的是空的字典
In [19]: color_set = {'green','blue','red','yellow','blue'}
In [20]: color_set
Out[20]: {'blue', 'green', 'red', 'yellow'}
可以看到，set()会帮你去掉重复的元素（上方的'blue'），下面讲一下集合的常见操作
判断一个元素是否在集合内
写法：element in set
In [21]: 'blue' in color_set
Out[21]: True
In [22]: 'white' in color_set
Out[22]: False
往集合中添加元素
写法：set.add(element)
In [23]: color_set.add('white')
In [24]: color_set
Out[24]: {'blue', 'green', 'red', 'white', 'yellow'}
# 若添加集合中已有的元素
# 则会自动去重
In [25]: color_set.add('blue')
In [26]: color_set
Out[26]: {'blue', 'green', 'red', 'white', 'yellow'}
移除元素
写法：set.remove(element)
In [27]: color_set.remove('white')
In [28]: color_set
Out[28]: {'blue', 'green', 'red', 'yellow'}
取集合的并集
写法：set_1.union(set_2)
In [29]: color_set1 = set(['green','blue','red','yellow','blue'])
In [30]: color_set2 = set(['purple','blue','pink','black'])
In [31]: color_set1.union(color_set2)
Out[31]: {'black', 'blue', 'green', 'pink', 'red', 'purple', 'yellow'}
并集.png
取集合的交集
写法：set_1.intersection(set_2)
In [29]: color_set1 = set(['green','blue','red','yellow','blue'])
In [30]: color_set2 = set(['purple','blue','pink','black'])
In [32]: color_set1.intersection(color_set2)
Out[32]: {'blue'}
交集.png
取集合的差集
写法：set_1.difference(set_2)
In [29]: color_set1 = set(['green','blue','red','yellow','blue'])
In [30]: color_set2 = set(['purple','blue','pink','black'])
# ①在color_set1中去掉color_set2含有的元素
In [33]: color_set1.difference(color_set2)
Out[33]: {'green', 'red', 'yellow'}
# ②在color_set2中去掉color_set1含有的元素
In [34]: color_set2.difference(color_set1)
Out[34]: {'black', 'pink', 'purple'}
差集1.png
差集2.png
练习题：
- 给定两个列表，分别为
[1, 2, 3, 3, 4, 4, 5]和[1, 1, 3, 5, 5, 7, 9]，请根据这两个列表分别生成集合A和B - 往集合A中加入元素
5和7，往集合B中加入元素6和9 - 求集合A和集合B的并集
- 求集合A和集合B的交集
- 求A-B和B-A
扫码关注公众号「Python读财」，第一时间获取干货，还可以加Python学习交流群！！
底部二维码.png
