1. 程式人生 > >Python3中多維list壓縮為一維,相當於Python2 中的flatten()

Python3中多維list壓縮為一維,相當於Python2 中的flatten()

import collections
def flatten(lst):
    for item in lst:
        if isinstance(item,collections.Iterable)and not isinstance(item,(str,bytes)):
            yield from flatten(item)
        else:
            yield item