展开列表
通过将列表的元素展开到一个新列表中,将列表展开。
- 遍历元素,如果元素是列表,则使用
list.extend()
,否则使用list.append()
。
def spread(arg):
ret = []
for i in arg:
ret.extend(i) if isinstance(i, list) else ret.append(i)
return ret
spread([1, 2, 3, [4, 5, 6], [7], 8, 9]) # [1, 2, 3, 4, 5, 6, 7, 8, 9]