Python中的集合介绍

小博测试成长之路

发布于 2023-09-01 08:22:59

1590

发布于 2023-09-01 08:22:59

文章被收录于专栏：软件测试学习软件测试学习

集合set是一个无序的、不可重复的元素集合。

集合的创建

大括号 {}
set()

a = {1, 2, 3, 4, 5}
print(type(a))

b = set([1,2,3,4,5,6,7])
print(type(b),b)

c = set((1,2,3))
print(c)

<class 'set'>
<class 'set'> {1, 2, 3, 4, 5, 6, 7}
{1, 2, 3}

集合推导式

跟列表和字典一样，集合也可以使用推导式进行生成

set1 = {x ** 2 for x in range(10)}
print(set1)  # 输出: {0, 1, 64, 4, 36, 9, 16, 49, 81, 25}

集合的更新操作

直接看下面的代码案例来掌握基本用法吧：

添加元素：

a = {1, 2, 3, 4, 5}
a.add(6)
print(a)  # {1, 2, 3, 4, 5, 6}

将一个或多个set合并到另一个set中：

set1 = {1, 2, 3}
set2 = {3, 4, 5}
set1.update(set2)
print(set1)  # {1, 2, 3, 4, 5}

set1 = {1, 2, 3}
set2 = {3, 4, 5}
set3 = {5, 6, 7}
set1.update(set2, set3)
print(set1)  # {1, 2, 3, 4, 5, 6, 7}

使用remove或者discard移除指定元素：

set1 = {1, 2, 3}
set1.discard(5)  # 从集合set1中移除元素5，如果元素不存在，不会引发错误
set1.remove(4)  # 从集合set1中移除元素4，如果元素不存在，会引发KeyError

注意：remove如果移除的元素不存在，会报错，discard不会报错。

集合的其他操作

并集：使用 union() 方法或 | 运算符来获取两个集合的并集。
交集：使用 intersection() 方法或 & 运算符来获取两个集合的交集。
差集：使用 difference() 方法或 - 运算符来获取一个集合相对于另一个集合的差集。
对称差集：使用 symmetric_difference() 方法或 ^ 运算符来获取两个集合的对称差集（即两个集合中不一致的元素）。

直接上代码，看效果：

set1 = {1, 2, 3}
set2 = {2, 3, 4}

union_set = set1.union(set2)  # 或者使用 union_set = set1 | set2
intersection_set = set1.intersection(set2)  # 或者使用 intersection_set = set1 & set2
difference_set = set1.difference(set2)  # 或者使用 difference_set = set1 - set2
symmetric_difference_set = set1.symmetric_difference(set2)  # 或者使用 symmetric_difference_set = set1 ^ set2

print(union_set,set1 | set2)  # 输出: {1, 2, 3, 4} {1, 2, 3, 4}
print(intersection_set, set1 & set2)  # 输出:{2, 3} {2, 3}
print(difference_set, set1 - set2)  # 输出: {1} {1}
print(symmetric_difference_set,set1 ^ set2)  # 输出: {1, 4} {1, 4}

集合中怎么判断一个集合是否是另一个集合的子集：

检查一个集合是否是另一个集合的子集:使用 issubset() 方法或 <= 运算符。
检查另一个集合是否是这个集合的子集:使用 issuperset() 方法或 >= 运算符。

可能看上面的文字有点抽象，什么这个集合那个集合的，接下来我们直接上案例，相信大家看了之后一目了然：

set1 = {1, 2, 3}
set2 = {1, 2, 3, 4}

print(set1.issubset(set2))  # 输出: True
print(set1 <= {1, 2, 3})  # 输出: True
print(set2.issuperset(set1))  # 输出: True
print(set2 >= set1)  # 输出: True

set3 = {4, 5, 6}

print(set1.issubset(set3))  # 输出: False
print(set3.issuperset(set1))  # 输出: False

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2023-05-19，如有侵权请联系 cloudcommunity@tencent.com 删除

python