列表生成式python_Python的列表生成式

列表生成式是Python中非常强大的可以用来创建list的生成式。

如果要生成list[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]，可以使用切片：

>>> range(1,11)

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

如果是[1*1,2*2,...,10*10]呢，可以使用循环：

>>> L = []

>>> for i in range(1,11):

... L.append(i*i)

...

>>> 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]

后面添加if语句进行筛选出仅偶数的平方：

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

[4, 16, 36, 64, 100]

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

>>> [m+n for m in 'ABC' for n in 'XYZ']

['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

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

>>> import os

>>> [d for d in os.listdir('.')]

['.DS_Store', '20101.py', '20903.py', 'Calculator.py', 'hello.py', 'learning.py', 'mymodule.py', 'tryExcept.py', 'tryExcept1.py']

for循环其实可以同时使用两个甚至多个变量，比如dict的iteritems()可以同时迭代key和value：

>>> d = {'x': 'A', 'y': 'B', 'z': 'C' }

>>> for j,k in d.iteritems():

... print j,'=',k

...

y = B

x = A

z = C

如果是使用列表生成式，那就简约多了：

>>> d = {'x': 'A', 'y': 'B', 'z': 'C' }

>>> [j+'='+k for j,k in d.iteritems()]

['y=B', 'x=A', 'z=C']

把一个list中的字符变成小写：

>>> L = ["Python","Java","Oc","Swift"]

>>> [s.lower() for s in L]

['python', 'java', 'oc', 'swift']

由于非字符串没有lower()的方法，如果list中含有非字符串，则调用lower()会报错：

>>> L = ["Python","Java",100,"Swift"]

>>> [s.lower() for s in L]

Traceback (most recent call last):

File "", line 1, in

AttributeError: 'int' object has no attribute 'lower'

因此，加上判断：

>>> L = ["Python","Java",100,"Swift"]

>>> [s.lower() for s in L if isinstance(s,str)]

['python', 'java', 'swift']

>>>