Python字典与集合推导式

引言

在Python中，推导式是一种简洁而强大的语法，用于快速创建列表、字典或集合。它们特别适合新人学习，因为它能让代码更简洁、易读。本教程将专注于字典推导式和集合推导式，帮助你从基础开始理解。

什么是字典推导式？

字典推导式允许你用一行代码生成字典。它的基本语法是：

{key_expression: value_expression for item in iterable}

或者可以加上条件过滤：

{key_expression: value_expression for item in iterable if condition}

简单示例

假设我们有一个数字列表，想创建一个字典，其中键是数字，值是数字的平方。

numbers = [1, 2, 3, 4, 5]
square_dict = {num: num**2 for num in numbers}
print(square_dict)  # 输出: {1: 1, 2: 4, 3: 9, 4: 16, 5: 25}

带条件的字典推导式

只选择偶数数字创建字典：

numbers = [1, 2, 3, 4, 5]
even_square_dict = {num: num**2 for num in numbers if num % 2 == 0}
print(even_square_dict)  # 输出: {2: 4, 4: 16}

什么是集合推导式？

集合推导式与列表推导式类似，但用于创建集合（无序且不重复）。语法是：

{expression for item in iterable}

或者带条件：

{expression for item in iterable if condition}

简单示例

从一个列表中提取唯一的偶数数字。

numbers = [1, 2, 2, 3, 4, 4, 5]
even_set = {num for num in numbers if num % 2 == 0}
print(even_set)  # 输出: {2, 4} （集合自动去重）

更多集合推导式示例

从字符串列表中获取长度大于3的单词集合：

words = ["apple", "bat", "cat", "dog", "elephant"]
long_words_set = {word for word in words if len(word) > 3}
print(long_words_set)  # 输出: {'apple', 'elephant'}

比较字典推导式和集合推导式

• 字典推导式：生成键值对字典，语法使用花括号 {} 和冒号 :。
• 集合推导式：生成唯一元素集合，语法使用花括号 {} 但没有冒号（或使用 set() 函数）。

两者都支持条件过滤，使得代码更灵活。

实用技巧

1. 保持简洁：推导式适用于简单场景；如果逻辑太复杂，考虑使用循环以提高可读性。
2. 注意性能：推导式通常高效，但在处理大数据时要注意内存使用。
3. 新手练习：多尝试编写示例代码，从简单到复杂逐步练习。

总结

字典推导式和集合推导式是Python中优雅的工具，可以大大简化代码。通过本教程的示例，你应该能够轻松入门。继续练习，你会越来越熟练！

如果你有任何问题，欢迎在评论区留言讨论。