python【5】-生成式，生成器

一、条件和循环

1. if语句

if <条件判断1>:<执行1>
elif <条件判断2>:<执行2>
else:<执行4>

例如：

age=5
if age>=18:print 'adult'
elif age>=6:print 'teenager'
else:print 'kid'

一定要注意条件语句结尾的冒号，以及代码段的缩进。

只要if参数是非零数值、非空字符串、非空list等，就判断为True，否则为False。

2.循环

range(num)：返回0到num-1之间的整数。例如：

print range(5)>>>[0,1,2,3,4]

for循环：for x in ...循环就是把每个元素代入变量x，然后执行缩进块的语句。

s=0
for i in range(101):s+=i
print s

while循环：只要条件满足，就不断循环，条件不满足时退出循环。

i=1
s=0while(i<=100):s=s+ii=i+1print s

3.迭代

dict类型for ..in ..迭代有以下几种方式：

d={'a':1,'b':2,'c':3}

默认迭代key

for k in d:print k

iterkeys()返回key迭代器

for key in d.iterkeys():print key

itervalues()返回value迭代器

for value in d.itervalues():print value

iteritems()返回键值对迭代器

for key,value in d.iteritems():print key+"="+str(value)

判断对象是否可迭代：通过collections模块的Iterable类型判断

from collections import Iterable
print isinstance(d,Iterable)

带下标迭代：Python内置的enumerate函数可以把一个list变成索引-元素对

l=range(5)
for i,value in enumerate(l):print i,value

二、生成式

1.列表生成式

列表生成式是python受欢迎的语法之一，通过一句简洁的语法就可以对一组元素进行过滤，还可以对得到的元素进行转换处理。语法格式为：

[exp for val in collection if condition]

相当于

result=[]
for val in collection:if(condition):result.append(exp)

我们看个例子：将列表中的字符串转换为小写组成新的列表。

L=['Hello',10,'World',None]
print [s.lower() for s in L if isinstance(s,str)]
运行结果：  ['hello', 'world']

2.字典生成式

字典生成式基本格式如下：

{key-exp:val-exp for value in collection if condition}

例如：

print {i:i*10 for i in range(1,10) if(i%2==0)}运行结果为： {8: 80, 2: 20, 4: 40, 6: 60}

3.集合生成式

集合生成式格式和列表生成式类似，只不过用的是大括号：

{exp for value in collection if condition}

例如：统计列表中字符串元素的各种长度。

L=['Hello',10,'World',None,'aa']
print {len(x) for x in L if isinstance(x,str)}运行结果： set([2, 5])

4.嵌套列表生成式

嵌套列表生成式要注意for循环的顺序。

例如：我们想提取由两个列表组成的嵌套列表中，含有a的字符串：

L1=[['Cathy','Li'],['Zhang','Wang','Mike','Tom','Jack']]
print [name for list in L1 for name in list  if name.count('a')>0]运行结果： ['Cathy', 'Zhang', 'Wang', 'Jack']

我们还可以将嵌套列表中的数据进行扁平化展现，例如：

L2=[(1,3,5),(2,4,6),(100,200)]
print [value for t in L2 for value in t]运行结果： [1, 3, 5, 2, 4, 6, 100, 200]

发现语法太简洁啦！！

三、生成器

在Python中，有一种自定义迭代器的方式，称为生成器（Generator）。

定义生成器的两种方式：

1.创建一个generator，只要把一个列表生成式的[]改成()，就创建了一个generator：

l=[x for x in range(1,10)]
print lg=(x for x in range(1,10))
print g.next()
print g.next()for x in g:print x运行结果：
[1, 2, 3, 4, 5, 6, 7, 8, 9]
1
2
3
4
5
6
7
8
9

generator保存的是算法，每次调用next()，就计算出下一个元素的值，直到计算到最后一个元素，没有更多的元素时，抛出StopIteration的错误。

2.定义generator的另一种方法。如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator：

def fib(n):a=1b=1i=0;yield ayield bwhile(i<n):a,b=b,a+bi+=1yield bfor x in fib(10):print x运行结果：
1
1
2
3
5
8
13
21
34
55
89
144

生成器和普通函数区别：

普通函数是顺序执行，遇到return语句或者最后一行函数语句就返回；

而变成generator的函数，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行。

def test():print 1yieldprint 2yieldprint 3yieldt=test()
t.next()#运行结果1
t.next()#运行结果2
t.next()#运行结果3

如需转载，请标明原文链接：http://www.cnblogs.com/janes/