python 集合常用操作汇总

时间:2022-12-25 野生大虾人气:0

无序、不重复、可迭代

demo = set()
demo.add(1)
>>>
{1}

demo.remove(1)
demo
>>>
set()

使用remove移除一个不存在的元素会抛出异常，使用以下方法则不会

demo.discard(1)
# 移除一个不存在的元素时，不会抛出异常 意味着并不知晓是否移除成功

s1 = {1, 3, 10, 2}
s2 = {2, 3}
s1 & s2
>>>
{2, 3}

s1 | s2
>>>
{1, 2, 3, 10}

s1 - s2
>>>
{1, 10}

# 如果s2 - s1 返回空集合 因为s2并不包含s1的任何元素
s2 - s1
>>>
set()

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('字符串')

加载全部内容