Python基础知识小结之集合

1. 集合是什么？

在Python中，集合是一种无序、唯一的数据结构，内部元素不允许重复，内部元素亦不支持索引操作。

1.1 创建集合

通过set()函数或者{}创建一个空集合：

s1 = set()
s2 = {}
print(type(s1))  # <class 'set'>
print(type(s2))  # <class 'dict'>

可以通过直接在{}内部加入元素来创建集合：

s = {1, 2, 3, 'a', 'b', 'c'}
print(s)  # {1, 2, 3, 'a', 'b', 'c'}

1.2 集合运算

集合支持基本的集合运算：

a = {1, 2, 3}
b = {2, 3, 4}
print(a | b)  # {1, 2, 3, 4}
print(a & b)  # {2, 3}
print(a - b)  # {1}
print(b - a)  # {4}

1.3 集合去重

可以通过将列表转换为集合来快速去重：

lst = [1, 2, 3, 2, 3, 4]
s = set(lst)
print(s)  # {1, 2, 3, 4}

2. 集合的方法

2.1 添加元素

可以使用add()方法向集合中添加一个元素，如果已存在则不做任何操作：

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

可以使用update()方法向集合中添加多个元素：

s = {1, 2, 3}
s.update([3, 4, 5])
print(s)  # {1, 2, 3, 4, 5}

2.2 删除元素

可以使用remove()方法删除集合中的一个元素：

s = {1, 2, 3}
s.remove(2)
print(s)  # {1, 3}

如果要删除的元素不存在，则会抛出KeyError异常。可以使用discard()方法删除集合中的一个元素，如果不存在则不做任何操作：

s = {1, 2, 3}
s.discard(2)
s.discard(4)
print(s)  # {1, 3}

2.3 集合方法示例

下面是一个完整的示例，演示了集合的使用方法：

s1 = {1, 2, 3}
s2 = {2, 3, 4}
s3 = s1 | s2
s4 = s1 & s2
s5 = s1 - s2
s6 = s2 - s1
s7 = [1, 2, 3, 2, 3, 4]
s8 = set(s7)
s9 = {5, 6, 7}
s9.add(8)
s9.update([9, 10, 11])
s10 = {1, 2, 3}
s10.remove(2)
s10.discard(4)
s10.discard(3)
print(s3)  # {1, 2, 3, 4}
print(s4)  # {2, 3}
print(s5)  # {1}
print(s6)  # {4}
print(s8)  # {1, 2, 3, 4}
print(s9)  # {5, 6, 7, 8, 9, 10, 11}
print(s10) # {1}

总结

集合是一种非常实用的数据结构，可以快速对数据进行去重、交集/并集/差集运算等。在实际开发中，也经常会用到集合。