Python集合函数：add()、remove()和union()

Python中的集合是一种无序，不重复的数据结构。它有很多有用的函数，包括add()、remove()和union()。本文将介绍这些函数的用法。

add()函数

add()函数用于向集合中添加元素。它的语法如下：

set.add(element)

其中set是要添加元素的集合，element是要添加的元素。例如，下面的代码向一个名为my_set的集合中添加了三个元素：

my_set = set()

my_set.add('apple')

my_set.add('banana')

my_set.add('grape')

remove()函数

remove()函数用于从集合中删除元素。它的语法如下：

set.remove(element)

其中set是要删除元素的集合，element是要删除的元素。例如，下面的代码从名为my_set的集合中删除了一个元素：

my_set = {'apple', 'banana', 'grape'}

my_set.remove('banana')

注意，如果要删除的元素不存在于集合中，remove()函数会引发一个KeyError异常。为避免这种情况发生，可以使用discard()函数。discrd()函数与remove()函数的用法相同，不同之处在于如果要删除的元素不存在的话，它不会引发异常而是什么也不做。

union()函数

union()函数用于将两个集合合并成一个新集合。它的语法如下：

set3 = set1.union(set2)

其中set1和set2是要合并的两个集合，set3是合并后得到的新集合。例如，下面的代码将两个集合合并成一个新集合：

set1 = {'apple', 'banana', 'grape'}

set2 = {'pear', 'orange', 'kiwi'}

set3 = set1.union(set2)

注意，union()函数不会修改原来的集合，而是创建一个新集合。如果要将多个集合合并成一个新集合，可以使用多个union()函数，如下所示：

set1 = {'apple', 'banana', 'grape'}

set2 = {'pear', 'orange', 'kiwi'}

set3 = {'watermelon', 'pineapple'}

set4 = set1.union(set2).union(set3)

这段代码将三个集合合并成一个新集合set4。

总结

add()函数用于向集合中添加元素，remove()函数用于从集合中删除元素，union()函数用于将多个集合合并成一个新集合。这些函数是Python集合操作中最常用的函数之一，掌握它们的用法非常重要。