Python学习--列表生成式

定义

列表生成式即List Comprehensions，是Python内置的非常简单却强大的，可以用来创建list的生成式。

在没有学习列表生成式前，要想生成list[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]，首先就想到list(range(1, 11))。

要想生成[1x1, 2x2, 3x3, …, 10x10]，首先想到循环
代码如下：

    L = []
    for i in range(1, 11):
        L.append(i * i)

那要是每次都想生成各种奇怪的list呢，每次都要重新写循环，重新定义函数，非常麻烦。这时列表生成式应运而生，并且一行解决。

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

一开始确实不习惯，写多了，其实也还好

各种写法

for循环后还可以加上if判断，这样就可以筛选出仅偶数的平方：

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

还可以使用两层循环，可以生成全排列：

L = [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

三层和三层以上就很少用到了

所以，运用列表生成式，可以写出非常简洁的代码。比如，列出当前目录下的所有文件和目录名 一行代码实现

import os
L = [d for d in os.listdir('.')]
['.emacs.d', '.ssh', '.Trash', 'Adlm', 'Applications', 'Desktop', 'Documents', 'Downloads', 'Library', 'Movies', 'Music', 'Pictures', 'Public', 'VirtualBox VMs', 'Workspace', 'XCode']

这个调用很好理解，listdir顾名思义，’.‘表示当前目录
列出当前目录下的文件和目录名

前面知道，for循环可以同时使用两个甚至多个变量，比如dict的items（），可以同时迭代key和value：

d = {'x': 'A', 'y': 'B', 'z': 'C'}
for k, v in d.items()
	print(k, ' = ', v)

y = B
x = A
z = C

所以列表生成式也可以使用两个变量来生成list：

d = {'x': 'A', 'y': 'B', 'z': 'c'}
L = [k + '=' + v for k, v in d.items()]
['y=B', 'x=A', 'z=C']

最后把一个list中所有的字符串变成小写：

L = ['Hello', 'World', 'IBM', 'Apple']
low = [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']

最后，对于列表生成式的if…else

使用列表生成式的时候，容易搞不清楚if…else的用法

例如， 下面代码正常输出偶数：

[x for x in range(1, 11) if x % 2 == 0]

但是不能在if后面加上else

>>> [x for x in range(1, 11) if x % 2 == 0 else 0]
  File "<stdin>", line 1
    [x for x in range(1, 11) if x % 2 == 0 else 0]
                                              ^
SyntaxError: invalid syntax

提示语法错误。
这是因为在for后面的if是一个筛选条件，不能带else，带了会增加歧义，无法筛选

那如果把if写在for前面，就必须加else， 否则报错

>>> [x if x % 2 == 0 for x in range(1, 11)]
  File "<stdin>", line 1
    [x if x % 2 == 0 for x in range(1, 11)]
                       ^
SyntaxError: invalid syntax

这是因为for前面的部分是一个表达式，它必须根据x计算出一个结果，因此，考察表达式：x if x % 2 == 0，它无法根据x计算结果，因为缺少else，所以必须加上else：

[x if x % 2 == 0 else -x for x in range(1, 11)]
[-1, 2, -3, 4, -5, 6, -7, 8, -9, 10]

所以，上面总结，在一个列表生成式中

• for前面的if…else是表达式
• for后面的if是过滤条件，不能带else

练习

如果list中既包含字符串，又包含整数，由于非字符串类型没有lower()方法，所以列表生成式会报错：

>>> L = ['Hello', 'World', 18, 'Apple', None]
>>> [s.lower() for s in L]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
AttributeError: 'int' object has no attribute 'lower'

使用内建的isinstance函数可以判断一个变量是不是字符串：

>>> x = 'abc'
>>> y = 123
>>> isinstance(x, str)
True
>>> isinstance(y, str)
False

请修改列表生成式，通过添加if语句保证列表生成式能正确地执行：

自然想到前置的if…else，可以生成全部

    L1 = ['Hello', 'World', 18, 'Apple', None]
    L2 = [x if not isinstance(x, str) else x.lower() for x in L1]
    print(L2)

或者后置的if,只能生成字符串

	L1 = ['Hello', 'World', 18, 'Apple', None]
	L2 = [x]