python学习笔记——列表生成式与生成器

1.列表生成式（List Comprehensions）

　　python中，列表生成式是用来创建列表的，相较于用循环实现更为简洁。举个例子，生成[1*1, 2*2, ... , 10*10]，循环用三行：

 L = []

 for i in range(1,11):

     L.append(i*i)

　　列表生成式只用一行，前面是生成规则，后面是初始元素，最后还可以加上判断条件：

 [i*i for i in range(1, 11)]

　　列表生成式还可以实现多层循环，以及判断，刚刚的栗子再写复杂一点就成了：

[a*b for a in range(1,11) for b in range(1,11) if a==b]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

2.生成器（Generator）

　　列表生成式生成的列表元素是有限的，因为列表毕竟要存储于内存中，而生成器不需要担心内存的问题。因为生成器保存的是算法，相较于列表生成式，生成器是时间换空间。

而生成器有两种创建方式：

　　2.1 将列表生成式的[]换成()：

 >>> g = (x * x for x in range(10))

 >>> g

 <generator object <genexpr> at 0x104feab40>

 >>> g.next()

 0

 >>> g.next()   #遍历也可以使用for循环，没有更多的元素时，抛出StopIteration的错误。

 1

　　2.2 利用函数实现，只需要把return语句变成yield即可，除此之外生成器是yield语句执行完后就停止，调用下一个再继续执行。

 def fib(max):

     n, a, b = 0, 0, 1

     while n < max:

         yield b

         a, b = b, a + b

         n = n + 1

巴特西

python学习笔记——列表生成式与生成器

最新文章

热门文章