問(wèn)題

將一個(gè)多層嵌套的序列展開(kāi)成一個(gè)單層列表

解決方案

可以寫(xiě)一個(gè)包含 yield from 語(yǔ)句的遞歸生成器來(lái)輕松解決這個(gè)問(wèn)題。比如：

from collections import Iterable

def 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 x


items = [1, 2, [3, 4, [5, 6], 7], 8]
for x in flatten(items):
    print(x, end = ' ')

示例代碼中，isinstance(x, Iterable) 檢查元素是否可迭代的。 如果是的話(huà)， yield from 就會(huì)返回所有子示例的元素。最終返回結(jié)果就是一個(gè)沒(méi)有嵌套的簡(jiǎn)單序列。

參數(shù) ignore_types 和檢測(cè)語(yǔ)句 isinstance(x, ignore_types) 用來(lái)將字符串和字節(jié)排除在可迭代對(duì)象外，防止將它們?cè)僬归_(kāi)成單個(gè)的字符。

討論

語(yǔ)句 yield from 在生成器中調(diào)用其他生成器作為子例程的時(shí)候非常有用。 如果不使用它的話(huà)，就必須寫(xiě)額外的 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

使用 yield from 語(yǔ)句，使得代碼更簡(jiǎn)潔。