python - 用列表和项目展平列表 : List comprehension with nested for loops and if else condition

Question

我想展平列表和项目的列表。以下代码仅适用于列表列表

def flatten(li):
return [item for sublist in li for item in sublist]

它会在

[[1],[2,3],[4]]

但不是

[[1],[2,3],[4],5]

我想要一个可以为上面的列表做的函数——比如

[item if isinstance(sublist, list) else sublist for sublist in li
for item in sublist]

以上代码报错

试穿

[[1],[2,3],[4],5]

给我

TypeError: 'int' object is not iterable

谁能给我一个没有的列表理解？

编辑:

我只想要一个列表综合。

Answer 1

def flatten(li):

return [ x for x in li if isinstance(x,int) ] + [item for sublist in li if isinstance(sublist,list) for item in sublist]