Python中的集合介绍

2023-09-01 08:22:59 浏览数 (2)

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

集合的创建

  • 大括号 {}
  • set()
代码语言:javascript复制
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}
  • 集合推导式

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

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

集合的更新操作

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

添加元素:

代码语言:javascript复制
a = {1, 2, 3, 4, 5}
a.add(6)
print(a)  # {1, 2, 3, 4, 5, 6}

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

代码语言:javascript复制
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移除指定元素:

代码语言:javascript复制
set1 = {1, 2, 3}
set1.discard(5)  # 从集合set1中移除元素5,如果元素不存在,不会引发错误
set1.remove(4)  # 从集合set1中移除元素4,如果元素不存在,会引发KeyError

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

集合的其他操作

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

直接上代码,看效果:

代码语言:javascript复制
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() 方法或 >= 运算符。

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

代码语言:javascript复制
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

0 人点赞