Python 集合推导式（Set Comprehension）-编程知识

Python 集合推导式（Set Comprehension）

news/2024/12/20 9:39:43/文章来源:https://www.cnblogs.com/liuyajun2022/p/18618437

Python 集合推导式（Set Comprehension） 是一种快速生成集合（set）的表达式。它的语法与列表推导式（List Comprehension）类似，但生成的是一个无序且不重复的集合。集合推导式使代码更加简洁、优雅，并能高效地处理需要去重的场景。

集合推导式的语法

{expression for item in iterable if condition}

expression：对每个元素进行处理的表达式。
item：从可迭代对象中提取的元素。
iterable：任何可以被迭代的对象，如列表、元组、字符串等。
condition（可选）：一个布尔表达式，用于过滤满足条件的元素。

基本示例

1. 简单集合生成

将列表中的元素生成一个集合：

nums = [1, 2, 3, 4, 5]
squares = {x**2 for x in nums}
print(squares)  # 输出：{1, 4, 9, 16, 25}

2. 去重功能

集合本身的特性可以自动去重：

nums = [1, 2, 2, 3, 4, 4, 5]
unique_nums = {x for x in nums}
print(unique_nums)  # 输出：{1, 2, 3, 4, 5}

3. 添加条件过滤

只生成满足条件的元素：

nums = [1, 2, 3, 4, 5]
even_squares = {x**2 for x in nums if x % 2 == 0}
print(even_squares)  # 输出：{4, 16}

高级示例

1. 处理字符串

从字符串中提取唯一字母（自动去重）：

text = "hello world"
unique_letters = {char for char in text if char.isalpha()}
print(unique_letters)  # 输出：{'h', 'e', 'l', 'o', 'w', 'r', 'd'}

2. 嵌套迭代

生成多个列表中元素的组合：

list1 = [1, 2, 3]
list2 = [4, 5, 6]
combinations = {x + y for x in list1 for y in list2}
print(combinations)  # 输出：{5, 6, 7, 8, 9}

3. 数据转换

从字典中提取值并去重：

student_scores = {'Alice': 90, 'Bob': 85, 'Charlie': 90}
unique_scores = {score for score in student_scores.values()}
print(unique_scores)  # 输出：{85, 90}

与列表推导式的区别

特性	集合推导式	列表推导式
生成的数据结构	`set`（集合）	`list`（列表）
元素是否重复	自动去重	可以包含重复元素
输出顺序	无序	有序
语法	使用 `{}`	使用 `[]`

性能比较

集合推导式 vs 手动生成集合

手动生成集合的代码通常需要更多步骤，而集合推导式更加简洁，同时性能相近。

示例：手动生成 vs 集合推导式

nums = [1, 2, 3, 4, 5]# 手动生成
result = set()
for x in nums:if x % 2 == 0:result.add(x**2)# 集合推导式
result_comprehension = {x**2 for x in nums if x % 2 == 0}

集合推导式的效率通常比循环+手动构造更高，且代码更易读。