Python 代码浏览合集介绍:为什么不举荐Python初学者间接看我的项目源码
本篇浏览的代码实现了将一个嵌套列表的所有嵌套档次齐全开展,造成一个简略的列表的性能。
本篇浏览的代码片段来自于30-seconds-of-python。
deep_flatten
from collections.abc import Iterabledef deep_flatten(lst): return [a for i in lst for a in deep_flatten(i)] if isinstance(lst, Iterable) else [lst]# EXAMPLESdeep_flatten([1, [2], [[3], 4], 5]) # [1,2,3,4,5]
deep_flatten
函数接管一个嵌套列表,返回齐全开展后的列表。函数应用isinstance()
与collections.abc.Iterable
来查看一个元素是否可迭代(是否为list
)。如果是,则在列表推导式内递归调用deep_flatten()
函数,否则返回[lst]
。
原函数:
def deep_flatten(lst): return [a for i in lst for a in deep_flatten(i)] if isinstance(lst, Iterable) else [lst]
能够改写成:
def deep_flatten(lst): if isinstance(lst, Iterable): return [a for i in lst for a in deep_flatten(i)] else: return [lst]
函数判断如果lst
是可迭代对象的话,就执行return [a for i in lst for a in deep_flatten(i)]
。这时如果i
是可迭代对象,在deep_flatten(i)
中就会持续调用该列表推导式,持续开展嵌套的列表;如果i
不是可迭代对象,在在deep_flatten(i)
中就会返回[i]
,此时a
的值就是i
,在列表推导式中就会失去一个非可迭代对象的元素,解开了元素上的嵌套层。
更近一步,函数能够改写成:
from collections.abc import Iterabledef deep_flatten(lst): temp = [] def f(lst): if isinstance(lst, Iterable): for i in lst: for a in f(i): temp.append(a) return [] else: return [lst] f(lst) return tempprint(deep_flatten([1, [2], [[3], 4], 5])) # [1,2,3,4,5]