集合的特性
无序、不重复、可迭代
常用api
- 创建一个集合
需要显式地使用set()
方法来声明,如果使用字面量{}
来声明解析器会认为这是一个字典。 - add()
往集合中添加一个元素
demo = set()
demo.add(1)
>>>
{1}
- 移除一个元素
demo.remove(1)
demo
>>>
set()
使用remove移除一个不存在的元素会抛出异常,使用以下方法则不会
demo.discard(1)
# 移除一个不存在的元素时,不会抛出异常 意味着并不知晓是否移除成功
特殊API
- 求交集
s1 = {1, 3, 10, 2}
s2 = {2, 3}
s1 & s2
>>>
{2, 3}
- 求并集
s1 | s2
>>>
{1, 2, 3, 10}
- 求差集 父-子原则 s1中包含了s2
s1 - s2
>>>
{1, 10}
# 如果s2 - s1 返回空集合 因为s2并不包含s1的任何元素
s2 - s1
>>>
set()
- 对称差集 返回不在s1和s2中的元素
有点拗口,可以简单理解为交集的取反
s1 ^ s2
>>>
{1, 10}
其他场景下的运用
- 判断某个序列是否包含多个指定对象
如,判断列表d = ['s', 't', 'a', 'r', 't']
中是否包含字符a
和r
,最常见的方式是来两个if判断,其实,也可以使用集合来完成:
d = ['s', 't', 'a', 'r', 't']
s = {'t', 'r'}
set(d) >= s # 列表d有s的元素
>>>
True
由此及彼,我们有可以借助集合来判断字典的key中是否包含多个指定key。
d = {
'name': 'mk',
'age': 20,
'job': None
}
d.keys() >= {'name', 'age'} # 字典d的keys中包含了集合的元素
>>>
True
当然,我们也可以更换比较运算符,如==
或者<=
,基本上可以断定,运算符两边的对象是可以调换的,无非是A是否包含B的逻辑判断。