Sometimes, we want to flatten a shallow list in Python.
In this article, we’ll look at how to flatten a shallow list in Python.
How to flatten a shallow list in Python?
To flatten a shallow list in Python, we can use the itertools.chain method.
For instance, we write:
import itertools
list_of_menuitems = [['image00', 'image01'], ['image10'], []]
chain = itertools.chain(*list_of_menuitems)
print(list(chain))
We defined the list_of_menuitems list which has lists inside it.
Then we call itertools.chain with the list_of_menuitems used as arguments since we spread it with *.
And then we convert the returned chain iterator to a list with list.
Therefore, we see:
['image00', 'image01', 'image10']
printed.
Conclusion
To flatten a shallow list in Python, we can use the itertools.chain method.