Python 如何展開嵌套的序列
問題
你想將一個多層嵌套的序列展開成一個單層列表
解決方案
可以寫一個包含 yield from 語句的遞歸生成器來輕松解決這個問題。比如:
from collections import Iterabledef flatten(items, ignore_types=(str, bytes)): for x in items: if isinstance(x, Iterable) and not isinstance(x, ignore_types): yield from flatten(x) else: yield xitems = [1, 2, [3, 4, [5, 6], 7], 8]# Produces 1 2 3 4 5 6 7 8for x in flatten(items): print(x)
在上面代碼中, isinstance(x, Iterable) 檢查某個元素是否是可迭代的。 如果是的話, yield from 就會返回所有子例程的值。最終返回結(jié)果就是一個沒有嵌套的簡單序列了。
額外的參數(shù) ignore_types 和檢測語句 isinstance(x, ignore_types) 用來將字符串和字節(jié)排除在可迭代對象外,防止將它們再展開成單個的字符。 這樣的話字符串?dāng)?shù)組就能最終返回我們所期望的結(jié)果了。比如:
>>> items = [’Dave’, ’Paula’, [’Thomas’, ’Lewis’]]>>> for x in flatten(items):... print(x)...DavePaulaThomasLewis>>>
討論
語句 yield from 在你想在生成器中調(diào)用其他生成器作為子例程的時候非常有用。 如果你不使用它的話,那么就必須寫額外的 for 循環(huán)了。比如:
def flatten(items, ignore_types=(str, bytes)): for x in items: if isinstance(x, Iterable) and not isinstance(x, ignore_types): for i in flatten(x):yield i else: yield x
盡管只改了一點(diǎn)點(diǎn),但是 yield from 語句看上去感覺更好,并且也使得代碼更簡潔清爽。
之前提到的對于字符串和字節(jié)的額外檢查是為了防止將它們再展開成單個字符。 如果還有其他你不想展開的類型,修改參數(shù) ignore_types 即可。
最后要注意的一點(diǎn)是, yield from 在涉及到基于協(xié)程和生成器的并發(fā)編程中扮演著更加重要的角色。
以上就是Python 如何展開嵌套的序列的詳細(xì)內(nèi)容,更多關(guān)于Python 展開嵌套的序列的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!
相關(guān)文章:
1. React實(shí)現(xiàn)一個倒計時hook組件實(shí)戰(zhàn)示例2. WMLScript的語法基礎(chǔ)3. CSS3實(shí)例分享之多重背景的實(shí)現(xiàn)(Multiple backgrounds)4. XML入門的常見問題(三)5. JavaScript中顏色模型的基礎(chǔ)知識與應(yīng)用詳解6. 不要在HTML中濫用div7. 報錯:XML頁無法顯示,下列標(biāo)記沒有被關(guān)閉解決方法8. el-input無法輸入的問題和表單驗(yàn)證失敗問題解決9. 前端html+css實(shí)現(xiàn)動態(tài)生日快樂代碼10. XML 增、刪、改和查示例
