Python集合推导式详解（零基础入门到实战应用）

什么是集合（Set）？

在Python中，集合（set）是一个无序、不重复的数据结构。它非常适合用于去重、成员检测和数学集合运算（如并集、交集等）。

# 创建一个普通集合my_set = {1, 2, 3, 3, 4}print(my_set)  # 输出: {1, 2, 3, 4} —— 自动去重

集合推导式的语法

集合推导式的基本语法如下：

{expression for item in iterable if condition}

• expression：对每个元素进行的操作或变换
• item：可迭代对象中的每一个元素
• iterable：任何可迭代对象（如列表、元组、字符串等）
• if condition（可选）：过滤条件，只保留满足条件的元素

基础示例：快速创建集合

假设我们有一个列表，想将其平方后转为集合（自动去重）：

# 普通方法numbers = [1, 2, 2, 3, 4, 4]squares = set()for n in numbers:    squares.add(n ** 2)print(squares)  # {16, 1, 4, 9}# 使用集合推导式squares = {n ** 2 for n in numbers}print(squares)  # {16, 1, 4, 9}

可以看到，集合推导式让代码更加简洁清晰，同时具备Python集合操作的天然优势——自动去重。

带条件过滤的集合推导式

你可以在推导式中加入 if 条件来筛选元素：

# 只保留偶数的平方numbers = [1, 2, 3, 4, 5, 6, 6]even_squares = {n ** 2 for n in numbers if n % 2 == 0}print(even_squares)  # {16, 4, 36}

与其他推导式的区别

Python还有列表推导式（[...]）和字典推导式（{key: value ...}），它们的区别在于：

• 列表推导式：用方括号 []，结果是列表，允许重复
• 集合推导式：用花括号 {}，结果是集合，自动去重且无序
• 字典推导式：用花括号但包含键值对 {k: v ...}

# 列表推导式 vs 集合推导式nums = [1, 1, 2, 3]list_comp = [x * 2 for x in nums]      # [2, 2, 4, 6]set_comp  = {x * 2 for x in nums}      # {2, 4, 6}

实用场景举例

1. 文本去重处理：从句子中提取唯一单词（忽略大小写）

sentence = "Hello world hello Python python"unique_words = {word.lower() for word in sentence.split()}print(unique_words)  # {'hello', 'world', 'python'}

2. 找出两个列表的共同元素（交集）

list1 = [1, 2, 3, 4, 5]list2 = [4, 5, 6, 7, 8]common = {x for x in list1 if x in list2}print(common)  # {4, 5}# 更高效的方式：直接用集合交集set1 = set(list1)set2 = set(list2)print(set1 & set2)  # {4, 5}

注意事项

• 集合是无序的，因此不能通过索引访问元素
• 集合中的元素必须是不可变类型（如数字、字符串、元组），不能是列表或字典
• 如果表达式返回可变对象（如列表），会报错：TypeError: unhashable type

# 错误示例try:    bad_set = {[1, 2] for x in range(1)}  # 报错！列表不可哈希except TypeError as e:    print(e)  # unhashable type: 'list'

总结

通过本教程，你应该已经掌握了Python集合推导式的核心用法。它不仅能让你的代码更简洁，还能自动处理去重问题，是Python集合操作中的强大工具。无论你是刚入门的Python初学者教程学习者，还是希望提升代码效率的开发者，掌握集合推导式都大有裨益。

记住关键点：

• 语法：{表达式 for 元素 in 可迭代对象 if 条件}
• 结果是无序、不重复的集合
• 适用于去重、筛选、转换等场景

继续练习，你很快就能熟练运用集合推导式语法写出优雅高效的Python代码！