1. 程式人生 > >python 64式: 第8式、生成器

python 64式: 第8式、生成器

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
關鍵:
1 生成器: 
含義: 一邊生成,一邊計算
優點: 適用於不能一下子將輸入資料儲存在列表中,然後遍歷輸入資料列表進行處理的場景
特點: 輸入的資料需要有某種推導或者計算規則
2 生成方式:
方式1: 圓括號
例如:
generator = (x * x for x in range(5))
列印元素:
for value in generator:
    print value

方式2: yield
def fibonacchi(num):
    counter = 0
    previous = 0
    current = 1
    while counter < num:
        yield current
        tmp = previous
        previous = current
        current += tmp
        counter += 1
執行原理:
生成器遇到yield返回,再次執行會從yield語句處繼續執行

3 應用:
ceilometer中獲取虛機列表中的cpu資料的時候用到生成器:
具體程式碼如下:
class CPUPollster(pollsters.BaseComputePollster):

    def get_samples(self, manager, cache, resources):
        for instance in resources:
            LOG.debug('checking instance %s', instance.id)
            try:
                cpu_info = self.inspector.inspect_cpus(instance)
                LOG.debug("CPUTIME USAGE: %(instance)s %(time)d",
                          {'instance': instance,
                           'time': cpu_info.time})
                cpu_num = {'cpu_number': cpu_info.number}
                yield util.make_sample_from_instance(
                    instance,
                    name='cpu',
                    type=sample.TYPE_CUMULATIVE,
                    unit='ns',
                    volume=cpu_info.time,
                    additional_metadata=cpu_num,
                    monotonic_time=monotonic.monotonic()
                )

總結:
生成器本質是邊迴圈邊計算,用於不能一下子將輸入資料儲存到列表而後遍歷列表進行處理的場景。
可以通過兩種方式得到生成器:1)圓括號, 即()生成,2)編寫的函式中,再獲取到處理結果的地方,將
return修改為yield即可。

參考:
https://www.liaoxuefeng.com/wiki/001374738125095c955c1e6d8bb493182103fac9270762a000/00138681965108490cb4c13182e472f8d87830f13be6e88000
'''

def fibonacchi(num):
    counter = 0
    previous = 0
    current = 1
    while counter < num:
        yield current
        tmp = previous
        previous = current
        current += tmp
        counter += 1

def process():
    for i in fibonacchi(5):
        print i

if __name__ == "__main__":
    process()