Python小记（八）：集合的基础操作入门

Python集合基础操作

1. 创建集合

创建集合的语法非常简单，使用大括号 {} 包裹元素即可。元素之间使用逗号 , 分隔。例如，以下代码创建了一个包含数字 1、2、3、4、5 的集合：

my_set = {1, 2, 3, 4, 5}

2. 添加元素

向集合中添加元素可以使用 add() 方法。add() 方法的参数是一个元素，将其添加到集合中。例如，以下代码向 my_set 集合中添加元素 6：

my_set.add(6)

3. 删除元素

从集合中删除元素可以使用 remove() 方法或 discard() 方法。remove() 方法的参数是一个元素，将其从集合中删除。如果要删除的元素不存在，则会引发 KeyError 异常。discard() 方法的参数也是一个元素，将其从集合中删除。如果要删除的元素不存在，则不会引发异常。

例如，以下代码从 my_set 集合中删除元素 3：

my_set.remove(3)

以下代码从 my_set 集合中删除元素 7，由于 7 不在集合中，因此不会引发异常：

my_set.discard(7)

4. 查找元素

判断某个元素是否在集合中可以使用 in 运算符。例如，以下代码判断元素 3 是否在 my_set 集合中：

if 3 in my_set:
    print("3 在集合中")
else:
    print("3 不在集合中")

5. 集合运算

集合之间可以使用并集、交集、差集、对称差集运算。

• 并集：集合 A 和集合 B 的并集是包含集合 A 和集合 B 中所有元素的新集合。可以使用 | 运算符或 union() 方法来计算并集。例如，以下代码计算集合 my_set 和集合 your_set 的并集：

result_set = my_set | your_set
result_set = my_set.union(your_set)

• 交集：集合 A 和集合 B 的交集是包含集合 A 和集合 B 中共同元素的新集合。可以使用 & 运算符或 intersection() 方法来计算交集。例如，以下代码计算集合 my_set 和集合 your_set 的交集：

result_set = my_set & your_set
result_set = my_set.intersection(your_set)

• 差集：集合 A 与集合 B 的差集是包含集合 A 中但不包含集合 B 中的元素的新集合。可以使用 - 运算符或 difference() 方法来计算差集。例如，以下代码计算集合 my_set 和集合 your_set 的差集：

result_set = my_set - your_set
result_set = my_set.difference(your_set)

• 对称差集：集合 A 与集合 B 的对称差集是包含集合 A 中但不包含集合 B 中的元素，以及包含集合 B 中但不包含集合 A 中的元素的新集合。可以使用 ^ 运算符或 symmetric_difference() 方法来计算对称差集。例如，以下代码计算集合 my_set 和集合 your_set 的对称差集：

result_set = my_set ^ your_set
result_set = my_set.symmetric_difference(your_set)

结语

集合是 Python 中非常重要的数据结构，掌握集合的基础操作可以帮助您轻松处理各种数据处理任务。