欢迎访问宙启技术站
智能推送

Python中常用的集合函数:add、remove、union等

发布时间:2023-05-28 07:29:41

Python中的集合是一种无序且 的数据结构,它的元素不会重复出现,因此在处理一些需要去重或者只需要关心元素是否存在的数据时,集合是一个非常有用的选择。Python提供了许多用于操作集合的内置函数和方法,包括添加新元素、删除元素、合并集合等功能。

1. add方法

add()方法用于向集合中添加新的元素,如果该元素已经在集合中存在,则add()方法会忽略该元素。例如:

s = set([1, 2, 3])
s.add(4)
print(s)    # 输出 {1, 2, 3, 4}

s.add(1)
print(s)    # 输出 {1, 2, 3, 4}

2. remove方法

remove()方法用于从集合中删除指定的元素,如果集合中不存在该元素,则remove()方法会抛出KeyError异常。例如:

s = set([1, 2, 3])
s.remove(2)
print(s)    # 输出 {1, 3}

s.remove(4) # 抛出KeyError异常

3. pop方法

pop()方法用于随机删除集合中的一个元素,并返回该元素。如果集合为空,则pop()方法会抛出KeyError异常。例如:

s = set([1, 2, 3])
x = s.pop()
print(x, s) # 输出 1 {2, 3}

s.pop()
s.pop()
s.pop()     # 抛出KeyError异常

4. union方法

union()方法用于合并两个集合,返回一个新的集合,其中包含两个集合中的所有不重复元素。例如:

s1 = set([1, 2, 3])
s2 = set([2, 3, 4])
s3 = s1.union(s2)
print(s3)   # 输出 {1, 2, 3, 4}

除了union()方法外,还有update()方法可以用于将一个集合中的元素添加到另一个集合中。

5. intersection方法

intersection()方法用于求两个集合的交集,返回一个新的集合,其中包含两个集合中都存在的元素。例如:

s1 = set([1, 2, 3])
s2 = set([2, 3, 4])
s3 = s1.intersection(s2)
print(s3)   # 输出 {2, 3}

6. difference方法

difference()方法用于求两个集合的差集,返回一个新的集合,其中包含存在于 个集合中但不存在于第二个集合中的元素。例如:

s1 = set([1, 2, 3])
s2 = set([2, 3, 4])
s3 = s1.difference(s2)
print(s3)   # 输出 {1}

7. symmetric_difference方法

symmetric_difference()方法用于求两个集合的对称差集,返回一个新的集合,其中包含存在于两个集合中但不存在于两个集合的交集中的元素。例如:

s1 = set([1, 2, 3])
s2 = set([2, 3, 4])
s3 = s1.symmetric_difference(s2)
print(s3)   # 输出 {1, 4}

总结

Python中的集合提供了一系列非常有用的内置函数和方法,包括添加元素、删除元素、合并集合、求交集、求差集、求对称差集等,可以大大简化集合操作的代码。此外,集合还支持一系列其他操作,如判断元素是否存在、求集合的长度等。在日常编程中,我们可以灵活运用这些集合函数和方法来完成各种操作。