列表生成式（第19篇）

你可以通过list(range(0,11))生成一个列表，但是如果让你生成[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]，那就比较困难，但是聪明的你想到了循环：

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

这样也可以，很棒。但是Python提供了一种更简单的方式，一行代码就解决：

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

这就是列表生成式。把需要生成的元素 i * i 放在前面，后面紧跟一个for循环，非常简单。
再看一个例子，将列表中所有的元素变成大写：

>>> L = ['Hello', 'World', 'IBM', 'Apple']
>>> [s.upper() for s in L]
['HELLO', 'WORLD', 'IBM', 'APPLE']

你还可以加上if语句来进行筛选：

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

这就相当于：

L = []
for i in range(0, 11):
    if (i * i) % 2 == 0:
        L.append(i * i)

不仅如此，列表生成式还可以实现多层循环：

>>> [m + n for m in '123' for n in 'abc']
['1a', '1b', '1c', '2a', '2b', '2c', '3a', '3b', '3c']

相当于：

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

利用列表生成式可以写出很简洁的代码。

列表生成式还可以用两个变量生成list：

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