Python必备技巧之集合Set的使用

什么是Set

Set是Python中的一种基本数据类型，类似于数学中的集合。在Set中，每个元素都是唯一的，不存在重复的元素。

Set的定义

使用set()函数可以创建一个空的Set，也可以使用{}中间加上元素集合的方式来定义Set，如下所示：

empty_set = set()
sample_set = {1, 2, 3, 4}

Set的操作

添加元素

使用add()方法可以向Set中添加元素，如下所示：

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

删除元素

使用remove()方法可以从Set中删除元素，如下所示：

sample_set = {1, 2, 3, 4, 5}
sample_set.remove(5)
print(sample_set)  # {1, 2, 3, 4}

求交集、并集、差集

可以使用&、|、-等符号来对Set进行求交集、并集、差集等操作，如下所示：

set1 = {1, 2, 3, 4}
set2 = {3, 4, 5, 6}
intersection_set = set1 & set2  # 求交集
print(intersection_set)  # {3, 4}

union_set = set1 | set2  # 求并集
print(union_set)  # {1, 2, 3, 4, 5, 6}

difference_set = set1 - set2  # 求差集
print(difference_set)  # {1, 2}

Set的应用示例

列表去重

通过Set的特性，我们可以很方便地对列表去重，如下所示：

lst = [1, 2, 3, 3, 4, 5, 5]
unique_lst = list(set(lst))
print(unique_lst)  # [1, 2, 3, 4, 5]

判断元素是否存在

Set中的元素是唯一的，因此我们可以使用Set来判断一个元素是否存在于集合中，如下所示：

sample_set = {1, 2, 3, 4}
if 1 in sample_set:
    print("1 exists in sample_set")
else:
    print("1 does not exist in sample_set")

总结

Set是Python中的一种基本数据类型，每个元素都是唯一的。我们可以使用Set来进行交集、并集、差集等操作，还可以用于列表去重、元素查找等场合。