Sentry Answers>Python>

Making a list out of a list of lists in Python?

Making a list out of a list of lists in Python?

James W.

—

The Problem

You want to flatten both regular and irregular lists of lists. Nested for loops could be used to solve this problem.

But are there any simpler solutions in Python?

The Solution

Using Python Libraries

The Python itertools module is a simple and readable method for flattening regular lists of lists:

Click to Copy
import itertools

regular_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flat_list = list(itertools.chain.from_iterable(regular_list))
print(flat_list)

Output:

Click to Copy
[1, 2, 3, 4, 5, 6, 7, 8, 9]

Using List Comprehension

Use list comprehension to flatten a regular list:

Click to Copy
regular_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flat_list = [item for inner_list in regular_list for item in inner_list]
print(flat_list)

Output:

Click to Copy
[1, 2, 3, 4, 5, 6, 7, 8, 9]

Using Nested Loops

Using nested loops might not be a shortcut, but this method is useful for flattening both regular and irregular lists of lists:

Click to Copy
def flatten_list(list_of_lists):
    flattened_list = []

    for element in list_of_lists:
        if type(element) is list:
            for item in element:
                flattened_list.append(item)
        else:
            flattened_list.append(element)
    return flattened_list

# Regular list of lists
regular_list_of_lists = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# Irregular list of lists
irregular_list_of_lists = [1, [2, 3, 4], 5, [6, 7], 8, 9]

print(flatten_list(regular_list_of_lists))
print(flatten_list(irregular_list_of_lists))

Output:

Click to Copy
[1, 2, 3, 4, 5, 6, 7, 8, 9]
[1, 2, 3, 4, 5, 6, 7, 8, 9]
  • Sentry BlogPython Performance Testing: A Comprehensive Guide
  • Syntax.fmListen to the Syntax Podcast
  • Sentry BlogLogging in Python: A Developer’s Guide
  • Syntax.fm logo
    Listen to the Syntax Podcast

    Tasty treats for web developers brought to you by Sentry. Get tips and tricks from Wes Bos and Scott Tolinski.

    SEE EPISODES

Considered “not bad” by 4 million developers and more than 100,000 organizations worldwide, Sentry provides code-level observability to many of the world’s best-known companies like Disney, Peloton, Cloudflare, Eventbrite, Slack, Supercell, and Rockstar Games. Each month we process billions of exceptions from the most popular products on the internet.

learn more about sentryjoin our discord
© 2024 • Sentry is a registered Trademark
of Functional Software, Inc.