Python編程如何在遞歸函數(shù)中使用迭代器
首先，想要實現(xiàn)的功能是遞歸遍歷文件夾，遇到滿足條件的文件時，用yield返回該文件的位置。
如果不用遞歸器，可以這樣實現(xiàn)：
path_list = []
def get_one_cage(root: str, cook_folder_name: str):
for item in os.listdir(root).copy():
item_path = os.path.join(root, item)
if item == cook_folder_name:
path_list.append(item_path)
return
elif os.path.isdir(item_path):
get_one_cage(item_path, cook_folder_name)
即，深度優(yōu)先遍歷，滿足要求時，將item_path補充到列表里，之后返回上一層。
這里有一個問題，需要有一個列表，把所有滿足條件的地址全存起來，占內(nèi)存。
使用迭代器可以用一個，遍歷出來一個，省內(nèi)存
替換為迭代器，最先想到的是，把 return 換成 yield，使用for循環(huán)調(diào)用迭代器函數(shù)
def get_one_cage(root: str, cook_folder_name: str):
for item in os.listdir(root).copy():
item_path = os.path.join(root, item)
if item == cook_folder_name:
yield item_path
elif os.path.isdir(item_path):
get_one_cage(item_path, cook_folder_name)
但是這樣的程序跑到內(nèi)嵌函數(shù)時，進不去，我百思不得其解
現(xiàn)在看，應該是因為迭代器函數(shù)不是一個函數(shù)，不是一個命令語句，它只是一個對象。
簡單說就是，python程序一般遵循：動詞+名詞的結構，或者動詞，比如：
a = 1
這句話實際上是把1賦值給了a，是有動詞的。
迭代器只是一個名詞，必須用for語句調(diào)用或者next()方法調(diào)用才會執(zhí)行，或者是print，yield，return等等，反正得加個動詞，不能孤零零一個名詞。
而且上述代碼還有一個漏洞。在第一段代碼中，我們用一個全局變量存放遍歷結果。在第二段代碼里，我們本意是把結果yield到for循環(huán)調(diào)用的地方，但事實是，程序已經(jīng)套了好幾層了，每次yiled只能返回一層。如下圖所示：
綜上兩點作出如下修改：
def get_one_cage(root: str, cook_folder_name: str):
for item in os.listdir(root).copy():
item_path = os.path.join(root, item)
if item == cook_folder_name:
yield item_path
elif os.path.isdir(item_path):
yield get_one_cage(item_path, cook_folder_name)
程序執(zhí)行結果如下：
顯然是返回了一個迭代器，不是一個str，其邏輯如下圖所示：
就好比，本意是：
小明把沙袋傳給小紅，小紅傳給小蘭
但現(xiàn)在是：
小明把沙袋傳給了小紅，小紅被傳了出去
修改如下：
def get_one_cage(root: str, cook_folder_name: str):
for item in os.listdir(root).copy():
item_path = os.path.join(root, item)
if item == cook_folder_name:
yield item_path
elif os.path.isdir(item_path):
yield next(get_one_cage(item_path, cook_folder_name))
邏輯如下：
還有一種情況是學長源碼里的：使用for調(diào)用迭代器：
def get_one_cage(root: str, cook_folder_name: str):
for item in os.listdir(root).copy():
item_path = os.path.join(root, item)
if item == cook_folder_name:
yield item_path
elif os.path.isdir(item_path):
for i in get_one_cage(item_path, cook_folder_name):
yield i
這使用于多個文件的返回，源碼里還配合isfile使用，這里是簡化版，所以顯得冗余。
兩種方式均可以正常使用。
昨天這篇文章寫完后，遇到了bug，簡單說就是，如果一個文件夾系統(tǒng)沒有我們想要的文件，遞歸到最深一層文件夾時，會報錯
1
可以理解為：老板讓員工找一樣東西，員工外包給編外人員。如果編外人員找到了想要的東西，一路傳遞回去，可以正常交差。如果沒找到，編外人員就會一直找，不停歇，找遍了所有能找到的地方（遍歷完整個文件夾）也沒能找到，就會報錯StopIteration。
因此，問題核心是，沒有一個返回機制。修改辦法是在遍歷最后加一個空返回
def get_one_cage(root: str):
for index, item in enumerate(os.listdir(root)):
item_path = os.path.join(root, item)
if item == 'cooked_xyz':
yield item_path
elif os.path.isdir(item_path):
yield next(get_one_cage(item_path))
elif index == len(os.listdir(root).copy()) - 1:
yield
或者是利用try… except語句處理異常:
def get_one_cage(root: str):
try:
for item in os.listdir(root):
item_path = os.path.join(root, item)
if item == 'cooked_xyz':
yield item_path
elif os.path.isdir(item_path):
yield next(get_one_cage(item_path))
except:
yield
會有如上報錯，正常。
最后的yield換成return也是可以的，但最好還是用yield，兩個混起來用怪怪的。
個人推薦第二種方法
注：copy()可以不用要
以上就是Python編程如何在遞歸中使用迭代器的詳細內(nèi)容，更多關于Python編程遞歸中使用迭代器的資料請關注腳本之家其它相關文章！
以上就是Python編程如何在遞歸函數(shù)中使用迭代器的詳細內(nèi)容，更多關于Python遞歸函數(shù)中使用迭代器的資料請關注腳本之家其它相關文章！
相關文章
Python使用matplotlib實現(xiàn)繪制自定義圖形功能示例
這篇文章主要介紹了Python使用matplotlib實現(xiàn)繪制自定義圖形功能,結合實例形式分析了Python基于matplotlib模塊實現(xiàn)自定義圖形繪制相關操作技巧,需要的朋友可以參考下2018-01-01
基于Python+Pygame實現(xiàn)變異狗大戰(zhàn)游戲
只有你想不到，沒有我找不到寫不了的好游戲！這篇文章就來和大家分享一下如何基于Python+Pygame實現(xiàn)變異狗大戰(zhàn)游戲，感興趣的可以了解一下2023-03-03
這篇文章主要介紹了Python中sorted如何自定義比較邏輯，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-02-02
這篇文章主要介紹了django實現(xiàn)日志按日期分割，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
從列表或字典創(chuàng)建Pandas的DataFrame對象的方法
這篇文章主要介紹了從列表或字典創(chuàng)建Pandas的DataFrame對象的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-07-07
python實現(xiàn)的登陸Discuz!論壇通用代碼分享
這篇文章主要介紹了python實現(xiàn)的登陸Discuz!論壇通用代碼分享,需要的朋友可以參考下2014-07-07
這篇文章主要為大家介紹了Python人工智能深度學習RNN的模型流程結構，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步2021-11-11
對于機器學習多分類模型來說，其評價指標除了精度之外，常用的還有混淆矩陣和分類報告，下面來展示一下如何繪制混淆矩陣，這在論文中經(jīng)常會用到。感興趣的可以了解一下2021-05-05
一個簡單的用戶注冊和登錄的頁面，涉及到驗證，數(shù)據(jù)庫存儲等等，本文主要介紹了Flask登錄注冊項目的簡單實現(xiàn)，從目錄結構開始，感興趣的可以了解一下2021-05-05
下面小編就為大家分享一篇python 實現(xiàn)在txt指定行追加文本的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04