python的生成器

python的生成器

可以把生成器理解成是特殊的迭代器。

生成器一次只能产生一个值，这样消耗的内存数量将大大减小，而且允许调用函数可以很快的处理前几个返回值，因此生成器看起来像是一个函数，但是表现得却像是迭代器。

例子

下面我就举几个例子：

一：迭代器和生成器

list = [x for x in range(10)]
print(type(list))
print(list)
it = iter(list)
while True:
    try:
        print(next(it), end='\t')
    except StopIteration:
        print('迭代结束')
        break
        
        
generator = (x for x in range(10))
print(type(generator))
print(generator)
while True:
    try:
        print(next(generator), end='\t')
    except StopIteration:
        print('迭代结束')
        break

输出结果

<class 'list'>
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
0	1	2	3	4	5	6	7	8	9	迭代结束
<class 'generator'>
<generator object <genexpr> at 0x000002840200B648>
0	1	2	3	4	5	6	7	8	9	迭代结束

​ 从代码和结果不难看出，生成器也是可以进行迭代的。并且生成器是一个类。

​ 生成器可以通过生成表达式生成。**generator = (x for x in range(10))返回的是一个生成器。通过生成器可以使用next()**来获取元素

二：yield字段

yield的作用是生成并返回一个值。

#普通的函数
def A():
    pass
print(type(A()))
print(A())


print('====================================')

#生成器
def MyGenerator():
    index = -1;
    ls = [0,1,2,3,4,5,6]
    while True:
        index += 1
        yield ls[index]
        
MyG = MyGenerator()
print(type(MyGenerator()), type(MyG))
print(MyGenerator(), MyG)
for i in range(7):
    print(next(MyGenerator()), next(MyG))

运行结果：

<class 'NoneType'>
None
====================================
<class 'generator'> <class 'generator'>
<generator object MyGenerator at 0x0000018D3F84B648> <generator object MyGenerator at 0x0000018D3F84B9C8>
0 0
0 1
0 2
0 3
0 4
0 5
0 6

从结果可以知道，一个函数也能写成生成器，只要函数里有yield字段，而yield类似于函数的return，作用也是放回一个值。这里值得注意的是**type(MyGenerator()), type(MyG)并不是同一个对象，这里的MyGenerator()**类对象虽然是生成器，但是并不能进行迭代。也就是说要使用生成器必须实例化先。

三：类里面的生成器

生成器也可以写到类的里面

class MyGeneratorClass:
    def __init__(self,*args):
        self.list=[x for x in args ]

    def generate(self):
        for index in range(len(self.list)):
            yield self.list[index]

G = MyGeneratorClass(1,2,3)
TG= G.generate()
print(type(G))
print(type(G.generate()), type(TG))
print(G.generate(), TG)
for i in range(3):
    print(next(G.generate()), next(TG))

运行结果：

<class '__main__.MyGeneratorClass'>
<class 'generator'> <class 'generator'>
<generator object MyGeneratorClass.generate at 0x00000237DBC5B648> <generator object MyGeneratorClass.generate at 0x00000237DBC5B9C8>
1 1
1 2
1 3

生成器也是可以写到类里面的，同样的要迭代就必须讲生成器实例化，这里的G虽然是MyGeneratorClass的实例化对象，但是其内部的生成器并没有实例化，所以G.generate()并不能进行迭代。

以上都是个人见解，有错请指出。