How to make a flat list out of a list of lists?

by

Given a list of lists t,

flat_list = [item for sublist in t for item in sublist]

which means:

flat_list = []
for sublist in t:
    for item in sublist:
        flat_list.append(item)

is faster than the shortcuts posted so far. (t is the list to flatten.)

Here is the corresponding function:

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

As evidence, you can use the timeit module in the standard library:

$ python -mtimeit -s't=[[1,2,3],[4,5,6], [7], [8,9]]*99' '[item for sublist in t for item in sublist]'
10000 loops, best of 3: 143 usec per loop
$ python -mtimeit -s't=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'sum(t, [])'
1000 loops, best of 3: 969 usec per loop
$ python -mtimeit -s't=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'reduce(lambda x,y: x+y,t)'
1000 loops, best of 3: 1.1 msec per loop

Explanation: the shortcuts based on + (including the implied use in sum) are, of necessity, O(T**2) when there are T sublists — as the intermediate result list keeps getting longer, at each step a new intermediate result list object gets allocated, and all the items in the previous intermediate result must be copied over (as well as a few new ones added at the end). So, for simplicity and without actual loss of generality, say you have T sublists of k items each: the first k items are copied back and forth T-1 times, the second k items T-2 times, and so on; total number of copies is k times the sum of x for x from 1 to T excluded, i.e., k * (T**2)/2.

The list comprehension just generates one list, once, and copies each item over (from its original place of residence to the result list) also exactly once.

More Related Contents:

  1. How do I make a flat list out of a list of lists?
  2. Matrix Transpose in Python [duplicate]
  3. What is the difference between flatten and ravel functions in numpy?
  4. Flatten an irregular list of lists
  5. How to define a two-dimensional array?
  6. Sort multidimensional array based on 2nd element of the subarray
  7. How to access the elements of a 2D array?
  8. How to flatten a tuple in python
  9. How to flatten a hetrogenous list of list into a single list in python? [duplicate]
  10. Flatten an irregular (arbitrarily nested) list of lists
  11. How to sort a 2D list? [closed]
  12. Converting List Comprehensions to For Loops in Python
  13. How can I turn a flat list into a 2D array in python?
  14. how to unpack a tuple in this form (‘abc’, [20, 20])? [closed]
  15. Access nested dictionary items via a list of keys?
  16. Writing a list to a file with Python
  17. Find the most common element in a list
  18. Finding median of list in Python
  19. Split a list into parts based on a set of indexes in Python
  20. Find first sequence item that matches a criterion [duplicate]
  21. How do I iterate through two lists in parallel?
  22. Unable to reverse lists in Python, getting “Nonetype” as list
  23. How to return a subset of a list that matches a condition [duplicate]
  24. Indexing a list with an unique index
  25. How do I split a string into a list of words?
  26. Why doesn’t assigning to the loop variable modify the original list? How can I assign back to the list in a loop? [duplicate]
  27. Mapping dictionary value to list
  28. How to change variables fed into a for loop in list form
  29. Length of longest word in a list
  30. list extend() to index, inserting list elements not only to the end

Leave a Comment