Python中的yield详解
閱讀別人的python源碼時碰到了這個yield這個關鍵字,各種搜索終于搞懂了,在此做一下總結:
它的缺陷是所有數據都在內存中,如果有海量數據的話將會非常耗內存。
代碼示例1
#encoding:UTF-8 def yield_test(n): for i in range(n): yield call(i) print("i=",i) #做一些其它的事情 print("do something.") print("end.") def call(i): return i*2 #使用for循環 for i in yield_test(5): print(i,",")結果是:
>>> 0 , i= 0 2 , i= 1 4 , i= 2 6 , i= 3 8 , i= 4 do something. end. >>>理解的關鍵在于:下次迭代時,代碼從yield的下一跳語句開始執行。
代碼示例2:
def node._get_child_candidates(self, distance, min_dist, max_dist):if self._leftchild and distance - max_dist < self._median:yield self._leftchildif self._rightchild and distance + max_dist >= self._median:yield self._rightchild與前面不同的是,這個函數中沒有for循環,但它依然可以用于迭代。
node._get_child_candidates函數中有yield,所以它變成了一個迭代器,可以用于迭代。
執行第一次迭代時(其實就是調用next()方法),如果有左節點并且距離滿足要求,會執行第一個yield,這時會返回self._leftchild并完成第一個迭代。
執行第二次迭代時,從第一個yield后面開始,如果有右節點并且距離滿足要求,會執行第二個yield,這時會返回self._rightchild并完成第一個迭代。
執行第三次迭代時,第二個yield后再無代碼,捕獲異常,退出迭代。
調用過程:
result, candidates = list(), [self] while candidates:node = candidates.pop()distance = node._get_dist(obj)if distance <= max_dist and distance >= min_dist:result.extend(node._values)candidates.extend(node._get_child_candidates(distance, min_dist, max_dist)) return result上面的node._get_child_candidates(self, distance, min_dist, max_dist)是放在extend()函數中作為參數的,為什么可以這么用,就因為extend函數的參數不僅僅支持array,只要它是一個迭代器就可以。它的原型是array.extend(iterable)。
作者:千若逸
鏈接:https://www.jianshu.com/p/d09778f4e055
來源:簡書
著作權歸作者所有。商業轉載請聯系作者獲得授權,非商業轉載請注明出處。
總結
以上是生活随笔為你收集整理的Python中的yield详解的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python中的虚拟环境-virtual
- 下一篇: Mybatis对象中含有list对象