python笔记-3.1章_列表生成式

#自学Python的学习笔记

列表生成式

列表生成式及List comprehension，是python内置的非常简单却强大的可以用来创建list的生成式

#生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
L  = list(range(1,11))
print(L)

 

#生成list[1x1, 2x2, 3x3, ..., 10x10]
L = [x*x for x in range(1, 11)]
print(L)

写列表生成式时，把要生成的元素x * x放到前面，后面跟for循环，就可以把list创建出来 

 

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

#for循环后面可以加上if判断，这样可以筛选出仅偶数的平方
L = [x*x for x in range(1, 11) if x%2 == 0]
print(L)

使用两层循环生成全排列

#使用两层循环生成全排列
L = [x + y for x in 'ABC' for y in 'abc']
print(L)

运用列表生成式，可以写出非常简洁的代码。

列出当前目录下的所有文件和目录名：

import os

L = [d for d in os.listdir()]
print(L)

 

for循环可以使用两个甚至多个变量，dict的items可以同时迭代key 和value：

d  = {'A':'a', 'B':'b', 'C':'c'}
L = [x + '='+ y for x, y in d.items()]
print(L)

 

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

#把一个list中的所有字符串变成小写
d = ['A', 'B', 'C', 'D']
L = [s.lower() for s in d]
print(L)

 

使用列表生成式的时候，不能在最后的if加上else，因为for后面的if是一个筛选条件，不能带else；但是if写在for前面必须加else，因为for前面的部分是一个表达式，它必须根据x计算出一个结果

 

可见，在一个列表生成式中，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语句保证列表生成式能正确地执行：

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

L2 = [s.lower() for s in L1 if isinstance(s ,str)]

print(L2)