推导式
格式
variable = [out_exp_res for out_exp in input_list if out_exp ==2]
# out_exp_res:列表生成元素表达式,可以是有返回值的函数
# for out_exp in input_list: 迭代input_list将out_exp传入out_exp_res
# if out_exp == 2:根据条件过滤
列表推导式
- 例一:30以内所有能被3整除的数
print([i for i in range(0,31) if i%3 == 0])
- 例二:30以内所有能被3整除的数的平方
print([i*i for i in range(30) if i % 3 == 0])
例三:找到嵌套列表中名字含有两个‘e’的所有名字
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
print([name for lst in names for name in lst if name.count('e') >= 2])
练习题
例1: 过滤掉长度小于3的字符串列表,并将剩下的转换成大写字母
例2: 求(x,y)其中x是0-5之间的偶数,y是0-5之间的奇数组成的元祖列表
例3: 求M中3,6,9组成的列表M = [[1,2,3],[4,5,6],[7,8,9]]
1.[name.upper() for name in names if len(name)>3] 2.[(x,y) for x in range(5) if x%2==0 for y in range(5) if y %2==1] 3. [row[2] for row in M]
生成器表达式
1.把列表推导式中的[]换成()得到的就是生成器表达式
2.列表解析与生成器表达式都是一种便利的编程方式,只不过生成器表达式更节省内存
3.Python不但使用迭代器协议,让for循环变得更加通用。大部分内置函数也是使用迭代器协议访问对象的。
例如:sum函数是Python的内置函数,该函数使用迭代器协议访问对象,而生成器实现了迭代器协议,所以,我们可以直接这样计算一系列值的和。
sum(x ** 2 for x in range(4))