展开列表

通过将列表的元素展开到一个新列表中,将列表展开。

  • 遍历元素,如果元素是列表,则使用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]