Python列表生成式详解：从基础到高级用法

Python列表生成式概述

介绍

本文将重点讲解Python语言中列表生成式（List Comprehension）的基础概念及其实际应用方法。

基础列表生成

生成列表

若需生成包含1到10整数的列表 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]，可直接使用 range(1, 11) 函数实现：

>>> range(1, 11)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

然而，如果需要生成每个元素为其平方值的列表 [1x1, 2x2, 3x3, ..., 10x10]，传统做法是使用循环：

>>> L = []
>>> for x in range(1, 11):
... L.append(x * x)
... 
>>> L
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

相比之下，列表生成式允许用单行代码替代上述繁琐的循环操作来生成相同的列表：

>>> [x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

这种简洁的语法结构被称为Python特有的列表生成式。借助它，开发者可以用极简的代码快速构建列表对象。

编写列表生成式时，核心逻辑是将期望生成的元素表达式（如 x * x）置于开头，随后跟随 for 循环语句。掌握这一结构后，创建列表变得十分高效，通过多次练习即可熟练运用该语法。

复杂表达式与字典迭代

复杂表达式 (TODO)

for 循环不仅可用于遍历普通列表，还能直接迭代字典（dict）对象。

假设存在如下字典数据结构：

d = { 'Adam': 95, 'Lisa': 85, 'Bart': 59 }

利用复杂的列表生成式，可以将其转换为标准的 HTML 表格格式：

tds = ['<tr><td>%s</td><td>%s</td></tr>' % (name, score) for name, score in d.iteritems()]
print '<table>'
print '<tr><th>Name</th><th>Score</th><tr>'
print '\n'.join(tds)
print '</table>'

注意：字符串格式化可使用 % 操作符，通过指定参数替换 %s 占位符。此外，字符串的 join() 方法能够将列表中的所有元素拼接为一个完整的字符串。

将上述代码输出的结果保存为 html 文件，即可在浏览器中查看渲染后的表格效果：

<table border="1">
<tr><th>Name</th><th>Score</th><tr>
<tr><td>Lisa</td><td>85</td></tr>
<tr><td>Adam</td><td>95</td></tr>
<tr><td>Bart</td><td>59</td></tr>
</table>

条件过滤机制

条件过滤

在列表生成式的 for 循环之后，还可以追加 if 判断语句以过滤数据。例如：

>>> [x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

若仅希望获取偶数的平方，在不修改 range() 参数的前提下，可通过添加 if 条件进行筛选：

>>> [x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]

当存在 if 条件时，只有判断结果为 True 的当前循环元素才会被添加到最终生成的列表中。

多层循环与全排列

多层表达式

for 循环支持嵌套结构，因此在列表生成式中，同样可以使用多层 for 循环来生成列表。 对于字符串 'ABC' 和 '123'，利用两层循环可以生成所有可能的字符组合（全排列）：

>>> [m + n for m in 'ABC' for n in '123']
['A1', 'A2', 'A3', 'B1', 'B2', 'B3', 'C1', 'C2', 'C3']

将其转换为常规循环代码的形式如下：

L = []
for m in 'ABC':
for n in '123':
L.append(m + n)

结语

以上即为Python列表生成式的核心内容详解，更多相关技术资料请查阅本站其他文章。