How can I use itertools to flatten a nested list in Python?

How can I use itertools to flatten a nested list in Python?

You can use itertools.chain.from_iterable() to flatten a nested list. Here's an example:

import itertools
nested_list = [[1, 2, 3], [4, 5], [6]]
flattened = list(itertools.chain.from_iterable(nested_list))
print(flattened)  # Output: [1, 2, 3, 4, 5, 6]

What if my nested list has different lengths?

The itertools.chain.from_iterable() method will still work regardless of the lengths of the inner lists. It will simply concatenate all elements in the order they appear.

Are there any other ways to flatten a list in Python?

Yes, you can also use list comprehensions or the numpy library if you're working with numerical data. For example:

flattened = [item for sublist in nested_list for item in sublist]

Can you explain how the list comprehension works?

Sure! The list comprehension iterates over each sublist in nested_list, and then over each item in sublist, collecting all items into a new list.