Python的列表推导式
Python的列表推导式(List Comprehension)是一种简洁高效的创建列表的方式,它通过单行表达式替代传统的for
循环,兼具可读性和性能优势。以下是核心要点与示例:
基本语法
[expression for item in iterable if condition]
-
expression
:对item
的计算或处理 -
iterable
:可迭代对象(如列表、字符串、range等) -
if condition
(可选):过滤条件
常见用法示例
生成平方数列表
squares = [x**2 for x in range(10)] # [0, 1, 4, ..., 81]
过滤偶数
evens = [x for x in range(20) if x % 2 == 0] # [0, 2, 4, ..., 18]
字符串处理
words = [word.upper() for word in ["hello", "world"]] # ["HELLO", "WORLD"]
嵌套循环
matrix = [[1, 2], [3, 4]] flattened = [num for row in matrix for num in row] # [1, 2, 3, 4]
字典推导式(类似语法)
squares_dict = {x: x**2 for x in [1, 2, 3]} # {1: 1, 2: 4, 3: 9}
性能与可读性
- 优势:比传统循环更快(C语言层优化),且更简洁
- 注意:复杂逻辑可能降低可读性,此时建议改用显式循环
适用场景
- 数据转换、过滤、聚合等操作
- 生成测试数据集或简单映射(如
[f"item_{i}" for i in range(5)]
)
列表推导式是Pythonic编程的重要工具,合理使用能显著提升代码效率与优雅度。