Python 集合备忘单

什么是集合？

集合是 Python 中的一种数据类型，它是一组无序、唯一的元素。集合可以进行交集、并集等运算。

创建集合

可以使用 {} 或 set() 函数来创建集合。

# 使用 {} 创建集合
set1 = {1, 2, 3}
print(set1)

# 使用 set() 函数创建集合
set2 = set([1, 2, 3])
print(set2)

输出结果：

{1, 2, 3}
{1, 2, 3}

请注意区分集合和字典的差异。使用 {} 创建时，如果没有指定键值对，则创建的是空字典。

# 错误的创建方法
set3 = {}
type(set3)   # 输出是 dict

常用操作

添加元素

使用 add() 方法来添加元素到集合中。

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

删除元素

使用 remove() 或 discard() 方法来删除集合中的元素。

set1 = {1, 2, 3}
set1.remove(2)
print(set1)   # 输出 {1, 3}

set1.discard(3)
print(set1)   # 输出 {1}

请注意，如果使用 remove() 方法来删除元素，而集合中没有该元素，则会抛出 KeyError 异常。而使用 discard() 则不会。

set1 = {1, 2, 3}
set1.remove(4)   # 抛出 KeyError 异常
set1.discard(4)  # 不会抛出异常

更新集合

使用 update() 方法来更新集合，将其他集合中的元素添加到当前集合中。

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

集合运算

可以使用交集、并集、差集等运算来操作集合。

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

# 交集
print(set1 & set2)   # 输出 {3, 4}

# 并集
print(set1 | set2)   # 输出 {1, 2, 3, 4, 5, 6}

# 差集
print(set1 - set2)   # 输出 {1, 2}

# 对称差集
print(set1 ^ set2)   # 输出 {1, 2, 5, 6}

判断元素是否在集合中

使用 in 或 not in 运算符来判断元素是否在集合中。

set1 = {1, 2, 3}
print(1 in set1)      # 输出 True
print(4 not in set1)  # 输出 True

总结

本文介绍了 Python 中集合的定义、创建、常用操作等内容，希望能够帮助大家更好地使用集合来处理数据。