python_生成器

一、列表生成式：

　　列表生成式即python内置的非常简单且强大的生成list的生成式。

　　range(1,10)方法可以生成一个简单的list，

l = []

for i in range(1,10):
    l.append(i)

print(l)
>>>[1, 2, 3, 4, 5, 6, 7, 8, 9]


l = []

for i in range(1,10):
    l.append(i*i)

print(l)
>>>[1, 4, 9, 16, 25, 36, 49, 64, 81]

 

　　  这样生成一个list有些许累赘，列表生成式就是为了简化这一过程：

l = [i for i in range(1,10)]
print(l)

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

L = [i * i for i in range(1,10)]
print(L)

>>>[1, 4, 9, 16, 25, 36, 49, 64, 81]

　　for后边还可以加if判断，过滤出列表中所有的奇数：

L = [i * i for i in range(1,10) if i % 2 != 0]
print(L)

>>>[1, 9, 25, 49, 81]

　　还可以两个for循环一起使用：

l = [ x * y for x in range(1,5) for y in range(6,10)]
print(l)

>>>[6, 7, 8, 9, 12, 14, 16, 18, 18, 21, 24, 27, 24, 28, 32, 36]

 

二、生成器（generator）：

　　生成器是一种只有在调用时才会开始计算生成相应的数据，没调用时内存中只存储相关数据的计算方式和当前数据的相关位置。

　　生成器可以大大节省我的内存空间。

 

　　创建生成器的方法有很多种，方法一：只需要把列表生成式的“[ ]”改成“( )”即可

l = [i for i in range(10)]
print(l)

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

g = (i for i in range(10))
print(g)

>>><generator object <genexpr> at 0x000001DBE98C9888>

　　生成器输出的是可迭代的内存对象，可以理解为计算数据的方法。那么如何输出生成器中的数据呢 ？

　　使用 next() 方法可以一个个的输出数据：

g = (i for i in range(10))
print(g)

print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))

>>><generator object <genexpr> at 0x000001C1102D9888>
    0
    1
    2
    3
    4

　　next()方法会把生成器中的数据一个个输出，若是next()的次数大于range()值的次数，则报StopIteration的错误。

　　显然，用next()方法有点太low了，既然生成器是可迭代对象，那正确的取值方法是for循环：

g = (i for i in range(5))

for i in g:
    print(i)

>>>0
    1
    2
    3
    4

 

　　像这样的如此潇洒与直接。

　　方法二：yield

def yrd():
    print("this is 1")
    yield 1
    print("this is 2")
    yield 5
    print("this is 3")
    yield 9

print(yrd())
>>><generator object yrd at 0x00000204F9609888>

y=yrd()
next(y)
>>>this is 1

next(y)
>>>this is 2

next(y)
>>>this is 3

 

　　　如果一个函数里有yield关键字，那个这个函数就不在是普通的函数了，而是摇身一变成为了生成器函数。

　　    生成器函数与普通的函数执行顺序是不一样的，普通的函数遇到return，或是执行到结尾就返回，而迭代器函数是每次调用next()方法时遇到yield就返回，再次执行next()方法时从上次yield的位置继续执行。

　　　同样，迭代器函数也可以使用for循环来取数据， for函数中其实也是封装的next()方法。当循环穷尽时for函数会自动获取报错退出。

def yrd():
    print("this is 1")
    yield 1
    print("this is 2")
    yield 5
    print("this is 3")
    yield 9

for y in yrd():
    print(y)

>>> this is 1
    1
    this is 2
    5
    this is 3
    9