【目錄】

一 、生成器和yield

1、什么是生成器

2、為何要有生成器 

3、如何使用生成器

二、yield運算式應用

三 、三元運算式、字典生成式、集合生成器、串列生成式、生成器運算式

 

一、生成器（generator）和yield

1、yield關鍵字

有了yield關鍵字，我們就有了一種自定義迭代器的實作方式，

yield可以用于回傳值，但不同于return，函式一旦遇到return就結束了，而yield可以保存函式的運行狀態掛起函式，用來回傳多次值

2、什么是生成器

 若函式體包含yield關鍵字，再呼叫函式，并不會執行函式體代碼，得到的回傳值即生成器物件

栗子：

>>> def my_range(start,stop,step=1):
...     print('start...')
...     while start < stop:
...         yield start
...         start+=step
...     print('end...')
... 
>>> g=my_range(0,3)
>>> g
<generator object my_range at 0x104105678>

生成器內置有__iter__和__next__方法，所以生成器本身就是一個迭代器

>>> g.__iter__
<method-wrapper '__iter__' of generator object at 0x1037d2af0>
>>> g.__next__
<method-wrapper '__next__' of generator object at 0x1037d2af0>

因而我們可以用next(生成器)觸發生成器所對應函式的執行：

>>> next(g) # 觸發函式執行直到遇到yield則停止,將yield后的值回傳，并在當前位置掛起函式
start...
0
>>> next(g) # 再次呼叫next(g)，函式從上次暫停的位置繼續執行，直到重新遇到yield...
1
>>> next(g) # 周而復始...
2
>>> next(g) # 觸發函式執行沒有遇到yield則無值回傳，即取值完畢拋出例外結束迭代
end...
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

for回圈迭代版：

>>> for i in countdown(3):
...     print(i)
... 
countdown start
3
2
1
Done!

二、yield運算式應用

（1）栗子1—— x=yield

def dog(name):
    print('道哥%s準備吃東西啦...' %name)
    while True:
        # x拿到的是yield接收到的值
        x = yield # x = '肉包子'
        print('道哥%s吃了 %s' %(name,x))


g=dog('alex')
g.send(None) # 等同于next(g)

g.send(['一根骨頭','aaa'])
# g.send('肉包子')
# g.send('一同泔水')
# g.close()
# g.send('1111') # 關閉之后無法傳值

（2）栗子2——  x=yield 回傳值 

def dog(name):
    food_list=[]
    print('道哥%s準備吃東西啦...' %name)
    while True:
        # x拿到的是yield接收到的值
        x = yield food_list # x = '肉包子'
        print('道哥%s吃了 %s' %(name,x))
        food_list.append(x) # ['一根骨頭','肉包子']

g=dog('alex')
res=g.send(None)  # next(g)
print(res)

res=g.send('一根骨頭')
print(res)

res=g.send('肉包子')
print(res)

（3）再來一顆栗子 

def func():
    print('start.....')
    x=yield 1111  # x='xxxxx'
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    yield 22222

g=func()
res=next(g)
print(res)

res=g.send('xxxxx')
print(res)

三 、三元運算式、串列生成式、字典生成式、集合生成器、生成器運算式

1、三元運算式

 三元運算式是python為我們提供的一種簡化代碼的解決方案，語法——

res = 條件成立時回傳的值 if 條件 else 條件不成立時回傳的值

栗子：

原版——

def max2(x,y):
    if x > y:
        return x
    else:
        return y

res = max2(1,2)

升級版——

x=1
y=2
res = x if x > y else y # 三元運算式

2、串列生成式

 串列生成式是python為我們提供的一種簡化代碼的解決方案，用來快速生成串列，語法如下：

[expression for item1 in iterable1 if condition1
for item2 in iterable2 if condition2
...
for itemN in iterableN if conditionN
]

#類似于
res=[]
for item1 in iterable1:
    if condition1:
        for item2 in iterable2:
            if condition2
                ...
                for itemN in iterableN:
                    if conditionN:
                        res.append(expression)

栗子：

原版——

egg_list=[]
for i in range(10):
    egg_list.append('雞蛋%s' %i)

升級版——

egg_list=['雞蛋%s' %i for i in range(10)]

應用——

l = ['alex_dsb', 'lxx_dsb', 'wxx_dsb', "xxq_dsb", 'egon']
new_l=[]
for name in l:
    if name.endswith('dsb'):
        new_l.append(name)


new_l=[name for name in l if name.endswith('dsb')]
new_l=[name for name in l]

print(new_l)

# 把所有小寫字母全變成大寫
new_l=[name.upper() for name in l]
print(new_l)

# 把所有的名字去掉后綴_dsb
new_l=[name.replace('_dsb','') for name in l]
print(new_l)

3、字典生成式

keys=['name','age','gender']
dic={key:None for key in keys}
print(dic)

items=[('name','egon'),('age',18),('gender','male')]
res={k:v for k,v in items if k != 'gender'}
print(res)

4、集合生成式

keys=['name','age','gender']
set1={key for key in keys}
print(set1,type(set1))

5、生成器運算式

創建一個生成器物件有兩種方式——

一種是呼叫帶yield關鍵字的函式，

另一種就是生成器運算式，與串列生成式的語法格式相同，只需要將[]換成()，即：

（expression for item in iterable if condition）

對比串列生成式回傳的是一個串列，生成器運算式回傳的是一個生成器物件：

>>> [x*x for x in range(3)]
[0, 1, 4]
>>> g=(x*x for x in range(3))
>>> g
<generator object <genexpr> at 0x101be0ba0>

>>> next(g)
0
>>> next(g)
1
>>> next(g)
4
>>> next(g) #拋出例外StopIteration

g=(i for i in range(10) if i > 3)
#！！！！！！！！！！！強調！！！！！！！！！！！！！！！
#此刻g內部一個值也沒有

print(g,type(g))

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

應用：

如果我們要讀取一個大檔案的位元組數，應該基于生成器運算式的方式完成

with open('db.txt','rb') as f:
    nums=(len(line) for line in f)
    total_size=sum(nums) # 依次執行next(nums)，然后累加到一起得到結果=

 詳解版：

with open('筆記.txt', mode='rt', encoding='utf-8') as f:
   #方式一：
    res=0
    for line in f:
        res+=len(line)
    print(res)

  # 方式二：
    res=sum([len(line) for line in f])
    print(res)

  #方式三 ：效率最高
    res = sum((len(line) for line in f))
  #上述可以簡寫為如下形式
    res = sum(len(line) for line in f)
    print(res)

標籤：Python

上一篇：【2020Python修煉記20】Python語法入門—迭代器

下一篇：Python切換版本工具pyenv