自学Python--迭代、列表生成式、生成器、迭代器

可迭代对象：可通过for循环遍历的对象。通过collections模块的Iterable判断

from collections import Iterable
print(isinstance('abc', Iterable)) # True

如果对列表实现类似java的下标循环，可通过内置函数enumerate

for i, v in enumerate([1, 2, 3, 4, 5, 6]):
    print(i, '-->', v)
'''
0 --> 1
1 --> 2
2 --> 3
3 --> 4
4 --> 5
5 --> 6
'''

 

列表生成式：

l1 = [x * x for x in range(1, 10) if x % 2 == 0]
print(l1) # [4, 16, 36, 64]

l2 = [m + n for m in 'ABC' for n in 'XYZ']
print(l2) # ['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

 

生成器generator：一边循环一边计算的机制

创建一个生成器：

把列表生成式的[]改成()即可，使用next()或者__next()__可获取生成器的下一个返回值：

l1 = (x * x for x in range(1, 10) if x % 2 == 0)
print(type(l1)) # <class 'generator'>
print(next(l1)) # 4
print(l1.__next__()) # 16

l2 = (m + n for m in 'ABC' for n in 'XYZ')
print(type(l2)) # <class 'generator'>

生成器是可迭代对象，因此也可以使用for循环遍历

 

斐波拉契数列：除前两个数外，任意一个属都是前两个数相加得到，如1，1，2，3，5，8，13，21

斐波拉契数列用列表生成式写不出来，但可以用函数打印出来：

def fib(max):
    a, b = 0 ,1
    for i in range(0, max):
        print(b)
        a, b = b, a + b
    return 'ok'

fib(8) # 1 1 2 3 5 8 13 21

 

把fib函数变成生成器generator：如果一个函数包含yield关键字，这个函数就是一个generator

def fib(max):
    a, b = 0 ,1
    for i in range(0, max):
        yield b
        a, b = b, a + b
    return 'ok'
res = fib(8)
for i in res:
    print(i) # 1 1 2 3 5 8 13 21

杨辉三角：todo

def triangles(n):
    current_list = [1]
    pre_list = [1 ,1]
    for i in range(1,n+1):
        if i == 1:
            yield [1]
        elif i == 2:
            yield [1, 1]
        else:
            
            for j in range(1,i-1):
                current_list.append(pre_list[j-1] + pre_list[j])
            current_list.append(1)
            yield current_list
            pre_list = current_list
            current_list = [1]

res = triangles(6)
for i in res:
    print(i)

 

迭代器：

可迭代对象：

可直接用for循环遍历的对象，如list、tuple、dict、set、str、generator

迭代器：

可以被next()函数调用并不断返回下一个值的对象称为迭代器Iterator

迭代器都是可迭代对象，但可迭代对象不一定是迭代器