Python——列表(list)推导式-Toy模板网

这篇具有很好参考价值的文章主要介绍了Python——列表(list)推导式。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

本文基于python3。

1、Python推导式

你可以理解成一种数据的处理方式。目的是为了构建一个新的数据序列。
Python 支持各种数据结构的推导式：

列表(list)推导式
字典(dict)推导式
集合(set)推导式
元组(tuple)推导式

各自从字面意思就知道，创建一个列表、字典、集合、元组。
我们本文只看列表(list)推导式。

2、列表(list)推导式

列表推导式创建列表的方式更简洁。
常见的用法：

对序列或可迭代对象中的每个元素应用某种操作，用生成的结果创建新的列表。
用满足特定条件的元素创建子序列。

2.1、定义

定义主要有两种：（该部分定义来自：菜鸟教程）

定义1：

[表达式 for 变量 in 列表] 
[out_exp_res for out_exp in input_list]

定义2

[表达式 for 变量 in 列表 if 条件]
[out_exp_res for out_exp in input_list if condition]

out_exp_res：列表生成元素表达式，可以是有返回值的函数。
for out_exp in input_list：迭代 input_list 将 out_exp 传入到 out_exp_res 表达式中。
if condition：条件语句，可以过滤列表中不符合条件的值。

总结：一个表达式，后面为一个 for 子句，然后跟着零个或多个 for 或 if 子句。结果是由表达式依据 for 和 if 子句求值计算而得出一个新列表。

基于总结话我们来实际操作一下。

2.2、实际操作

2.2.1、一个表达式，后面为一个 for 子句

创建数值*2 的列表：

TestList = [x*2 for x in range(10)]

print(TestList)
# [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

等价于以下语句：

TestList = []
for x in range(10):
    TestList.append(x*2)

print(TestList)

2.2.2、一个表达式，后面为一个 for 子句，然后，跟着if 子句。

创建全是偶数的列表：

TestList = [i for i in range(10) if i % 2 == 0]

print(TestList)
# [0, 2, 4, 6, 8]

等价于以下语句：

TestList = []
for i in range(10):
    if i % 2 == 0:
        TestList.append(i)

print(TestList)

2.2.3、一个表达式，后面为一个 for 子句，然后，跟着 for 子句。

直接上实例，一看就懂：

OldList = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

TestList = [j for i in OldList for j in i]

print(TestList)
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

等价于以下语句：

OldList = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
TestList = []

for i in OldList:
    for j in i:
        TestList.append(j)

print(TestList)
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

2.2.4、一个表达式，后面为一个 for 子句，然后，跟着 for 子句，if 语句

直接上实例：

OldList = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

TestList = [j for i in OldList for j in i if j % 2 == 0]

print(TestList)
# [2, 4, 6, 8]

等价于以下语句：

OldList = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
TestList = []

for i in OldList:
    for j in i:
        if j % 2 == 0:
            TestList.append(j)

print(TestList)
# [2, 4, 6, 8]

2.2.5、表达式的不同

因为列表生成元素表达式，可以是有返回值的函数。因此就有一些其他用法。

2.2.5.1、表达式是函数

OldList = ['   AA', '  BB ', 'CC  ']
TestList = [i.strip() for i in OldList]
# strip()：去掉前后空格

print(TestList)
# ['AA', 'BB', 'CC']

2.2.5.2、表达式是元组

TestList = [(x, x**2) for x in range(6)]

print(TestList)
# [(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]

注意：使用元组必须带小括号。

# 没带小括号
TestList = [x, x**2 for x in range(6)]
print(TestList)

# 错误信息
/Scripts/python.exe d:/05.ProjectWork/00.Python/00.Project/test_python/test.py
  File "d:/05.ProjectWork/00.Python/00.Project/test_python/test.py", line 192
    TestList = [x, x**2 for x in range(6)]
                          ^
SyntaxError: invalid syntax

2.2.5.2、表达式是复杂的表达式和嵌套函数

from math import pi
TestList = [str(round(pi, i)) for i in range(1, 9)]

print(TestList)
# ['3.1', '3.14', '3.142', '3.1416', '3.14159', '3.141593', '3.1415927', '3.14159265']

3、利用列表推导式实现乘法口诀

直接上代码：

print("\n".join([" ".join(["{}*{}={}".format(x, y, x*y) for x in range(1,y+1)]) for y in range(1,10)]))

结果如下：

1*1=1
1*2=2 2*2=4
1*3=3 2*3=6 3*3=9
1*4=4 2*4=8 3*4=12 4*4=16
1*5=5 2*5=10 3*5=15 4*5=20 5*5=25
1*6=6 2*6=12 3*6=18 4*6=24 5*6=30 6*6=36
1*7=7 2*7=14 3*7=21 4*7=28 5*7=35 6*7=42 7*7=49
1*8=8 2*8=16 3*8=24 4*8=32 5*8=40 6*8=48 7*8=56 8*8=64
1*9=9 2*9=18 3*9=27 4*9=36 5*9=45 6*9=54 7*9=63 8*9=72 9*9=81