掃二維碼與項目經(jīng)理溝通
我們在微信上24小時期待你的聲音
解答本文疑問/技術(shù)咨詢/運營咨詢/技術(shù)建議/互聯(lián)網(wǎng)交流
我們經(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
}推薦學(xué)習(xí)《Python視頻教程》
你肯定想到了使用遞歸來解決這個問題,那么你可以試一試,看看你的遞歸函數(shù)有多少行代碼。
今天,我們使用yield關(guān)鍵字來實現(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)鍵字來實現(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)運行結(jié)果如下圖
通過使用 yield關(guān)鍵字,字典的key會像是在流水線上一樣,一層一層從內(nèi)向外進行組裝,從而形成完整的路徑。

我們在微信上24小時期待你的聲音
解答本文疑問/技術(shù)咨詢/運營咨詢/技術(shù)建議/互聯(lián)網(wǎng)交流