如何使用Python的集合函数?
发布时间:2023-06-21 15:02:11
集合(set)是一种基于散列表的数据类型,它的元素是无序的、 的。Python中有许多与集合相关的函数,这些函数可以简化集合的操作。
创建集合
可以使用花括号 {} 或 set() 函数来创建一个集合。
a = {'apple', 'banana', 'cherry'}
print(a)
b = set(['apple', 'banana', 'cherry'])
print(b)
输出:
{'banana', 'cherry', 'apple'}
{'banana', 'cherry', 'apple'}
可以看到,集合中的元素是无序的,且每个元素只出现一次。
添加元素
可以使用 add() 函数来向集合中添加一个元素,若元素已经存在,则不做任何操作。
a = {'apple', 'banana', 'cherry'}
a.add('orange')
print(a)
a.add('banana')
print(a)
输出:
{'orange', 'cherry', 'banana', 'apple'}
{'orange', 'cherry', 'banana', 'apple'}
移除元素
可以使用 remove() 或 discard() 函数来移除集合中的一个元素。
remove() 函数会报错,如果集合中不存在要移除的元素。而 discard() 函数不会报错,即使元素不存在也不会有任何影响。
a = {'apple', 'banana', 'cherry'}
a.remove('banana')
print(a)
a.discard('orange') # 不会报错
print(a)
a.remove('orange') # 会报错
输出:
{'cherry', 'apple'}
{'cherry', 'apple'}
KeyError: 'orange'
集合间的运算
交集、并集、差集、对称差集等运算可以用如下函数实现:
a = {1, 2, 3}
b = {2, 3, 4}
print(a & b) # 交集
print(a | b) # 并集
print(a - b) # 差集
print(a ^ b) # 对称差集
输出:
{2, 3}
{1, 2, 3, 4}
{1}
{1, 4}
其他集合函数
Python中还有许多其他与集合相关的函数,例如:
- len(s):返回集合元素个数。
- s.copy():返回集合的浅复制。
- s.clear():移除集合中的所有元素。
- s.pop():随机移除集合中的一个元素,若集合为空则报错。
- s.update(x):将集合s与可迭代对象x中的元素合并。
- s.intersection_update(x)、s.difference_update(x)、s.symmetric_difference_update(x):分别对集合s进行交集、差集、对称差集运算,并更新集合s。
总结
Python的集合函数能够使集合的操作更为简便和高效,其中包括创建集合、添加和移除元素、集合间的运算等。掌握集合函数的使用可以让Python程序变得更加简洁和清晰。
