RosettaCodeData/Task/Flatten-a-list/Python/flatten-a-list-3.py

13 lines
283 B
Python
Raw Permalink Normal View History

2020-02-17 23:21:07 -08:00
>>> def flatten(lst):
for x in lst:
if isinstance(x, list):
for x in flatten(x):
yield x
else:
yield x
2013-04-10 21:29:02 -07:00
2015-02-20 00:35:01 -05:00
>>> lst = [[1], 2, [[3,4], 5], [[[]]], [[[6]]], 7, 8, []]
2020-02-17 23:21:07 -08:00
>>> print list(flatten(lst))
2013-04-10 21:29:02 -07:00
[1, 2, 3, 4, 5, 6, 7, 8]