python 集合常用操作

集合的特性

无序、不重复、可迭代

常用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的逻辑判断。

注意

使用set声明集合时，传入的对象是一个可迭代对象，如：

set(['a', 'b'])

这个时候需要注意的是，如果要保存一个字符串到集合中，万不可使用：

set('字符串')

这样的方式，该方法会将字符串遍历逐个添加到集合中，导致最后的结果是{'字', '符', '串'}而非预期的{'字符串'}，要达到预期，可使用：

set(['字符串'])
set().add('字符串')