Python技巧——列表生成式

◤一、引言◢

今天为大家介绍一个python使用技巧——列表生成式。列表生成式是用一行代码快速制作列表的方法，在进行程序编程、调测时可以快速制作需要的测试数据。有时候也叫列表推导式，一个意思，大家不要纠结于名字。

◤二、if语句列表生成式◢

if语句的列表生成式类似C的三元运算符?:，但其输出的是一个元素的列表。

>>> #判断 a，b的大小>>> a = 10>>> b = 20>>> # if的常规语法>>> if a >= b :...     print('{0} 大于等于 {1}'.format(a,b))... else:...     print('{0} 小于 {1}'.format(a,b))...10 小于 20>>> # if的列表生成式语法>>> c = [print('{0} 大于等于 {1}'.format(a,b)) if a > b else print('{0} 小于 {1}'.format(a,b))]10 小于 20>>> c[None]

最后返回值赋给变量c，内容为None。这是因为print()函数的返回值是None，如果我们替换为其他的值，则会相应的显示。但要注意，这种方法输出的是一个列表，if语句的列表生成式输出的是一个元素的列表，实际应用较少。

if的列表生成式语法结构：

◤三、for语句的列表生成式◢

for语句的列表生成式是较为常见的应用，可以用一条语句生成复杂的列表数据。

# 使用*画一个直角三角形，共计5层，逐层递加。>>> # for的常规语法>>> for i in range(5):...     print('{0: <10}'.format('*'*(i+1)))...***************>>> # for的列表生成式语法>>> c = [print('{0: <10}'.format('*'*(i+1))) for i in range(5)]***************>>> c[None, None, None, None, None]

上述这个for的列表生成式返回值赋值给变量c，内容为 5个元素的列表，每个元素均为None。这是因为print()函数的返回值是None，下面我们改为输出i。

>>> c = [i for i in range(5)]>>> c[0, 1, 2, 3, 4]

可以看到，修改后变量c的值变为了列表[0, 1, 2, 3, 4]。for的列表生成式可以快速制作一个满足要求的列表数据，比较常用。而且它还可以嵌套if语句、lambda函数或自定义函数，实现更多元的列表。

for的列表生成式语法结构：

对于上述的列表生成式，语句块部分是可以嵌套更多的语句的，如继续嵌套for的列表生成式、if的列表生成式，或者其它的。

◤四、总结◢

本节所讲述的列表生成式方法也适合生成器的制作，方法是把方括号替换为圆括号，有兴趣的小伙伴可以自行实验一下。生成器可以较好地控制内存占用问题，对内存不足的场景较为适应。

下面是生成器和列表内存占用的一个示例：

生成器仅占用200个字节，列表占用了920个字节，随着数据量的增加差异会更大。这主要是因为生成器仅在需要时生成数据，不像列表是一次性把所有的数据都记录到内存中。

-----------------------------