我有以下代码,我需要它使用yield来解压缩列表:
def flat_list(array):
d=[]
for i in array:
if not isinstance(i, list):
yield i
else:
flat_list(i)
Run Code Online (Sandbox Code Playgroud)
例如,flat_list([1, [2, 2, 2], 4])应该返回[1, 2, 2, 2, 4]。我的代码返回[1, 4]。
从上一个问题中我了解到,我不仅需要递归调用一个函数,而且还需要指定它应该做什么。
但是如何将flat_list(i)项目添加到yield项目中?有点像flat_list(i).extend(yield i)。