python使用yield壓平嵌套字典的超簡單方法
我們經(jīng)常遇到各種字典套字典的數(shù)據(jù),例如:
nest_dict = { 'a': 1, 'b': { 'c': 2, 'd': 3, 'e': {'f': 4} }, 'g': {'h': 5}, 'i': 6, 'j': {'k': {'l': {'m': 8}}} }
有沒有什么簡單的辦法,把它壓扁,變成:
{ 'a': 1, 'b_c': 2, 'b_d': 3, 'b_e_f': 4, 'g_h': 5, 'i': 6, 'j_k_l_m': 8 }
你肯定想到了使用遞歸來解決這個問題,那么你可以試一試,看看你的遞歸函數(shù)有多少行代碼。
今天,我們使用yield關(guān)鍵字來實(shí)現(xiàn)這個需求,在不炫技的情況下,只需要8行代碼。在炫技的情況下,只需要3行代碼。
要快速地把這個嵌套字典壓扁,我們需要從下網(wǎng)上來處理字段。例如對于b->e->f->4這條路徑,我們首先把最里面的{'f': 4}轉(zhuǎn)換為一個元組('f', 4)。然后,把這個元組向上拋出,于是得到了元組('e', ('f', 4))。我們把 e拼接到f的前面,變?yōu)椋?'e_f', 4),繼續(xù)往上拋出,得到('b', ('e_f', 4))。再把b拼接到e_f上面,得到('b_e_f', 4)。完成一條線路的組裝。
這個邏輯如果使用yield關(guān)鍵字來實(shí)現(xiàn),就是:
def flat(x): for key, value in x.items(): if isinstance(value, dict): for k, v in flat(value): k = f'{key}_{k}' yield (k, v) else: yield (key, value)
運(yùn)行結(jié)果如下圖所示:
通過使用 yield關(guān)鍵字,字典的key會像是在流水線上一樣,一層一層從內(nèi)向外進(jìn)行組裝,從而形成完整的路徑。
在下一篇文章中,我們繼續(xù)使用yield關(guān)鍵字來解決字典與列表混合嵌套的情況。
總結(jié)
以上就是這篇文章的全部內(nèi)容了,希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,謝謝大家對腳本之家的支持。
相關(guān)文章
python matplotlib工具欄源碼探析三之添加、刪除自定義工具項(xiàng)的案例詳解
這篇文章主要介紹了python matplotlib工具欄源碼探析三之添加、刪除自定義工具項(xiàng)的案例詳解,本文通過實(shí)例代碼給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-02-02Python調(diào)用REST API接口的幾種方式匯總
這篇文章主要介紹了Python調(diào)用REST API接口的幾種方式匯總,幫助大家更好的利用python進(jìn)行自動化運(yùn)維,感興趣的朋友可以了解下2020-10-10Python實(shí)現(xiàn)定時精度可調(diào)節(jié)的定時器
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)定時精度可調(diào)節(jié)的定時器,具有一定的參考價值,感興趣的小伙伴們可以參考一下2018-04-04