Python | Sort Flatten list of list
The flattening of list of lists has been discussed earlier, but sometimes, in addition to flattening, it is also required to get the string in a sorted manner. Let’s discuss certain ways in which this can be done.
Method #1 : Using sorted() + list comprehension This idea is similar to flattening a list of list but in addition to it, we add a sorted function to sort the returned flattened list done by list comprehension.
- Python3
Python3
# Python3 code to demonstrate# sort flatten list of list # using sorted + list comprehension# initializing list of list test_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of list print("The original list : " + str(test_list))# using sorted + list comprehension# sort flatten list of listres = sorted([j for i in test_list for j in i])# print resultprint("The sorted and flattened list : " + str(res)) |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list : [1, 3, 3, 5, 7, 9, 12]
Time Complexity: O(n log n) where n is the total number of elements in the nested list.
Auxiliary Space: O(n)
Method #2 : Using itertools.chain() + sorted() The task that was done by list comprehension above can also be performed using the chain function that links elements of list and then sorted function does the task of sorting.
- Python3
Python3
# Python3 code to demonstrate# sort flatten list of list # using itertools.chain() + sorted()from itertools import chain# initializing list of list test_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of list print("The original list : " + str(test_list))# using itertools.chain() + sorted()# sort flatten list of listres = sorted(chain(*test_list))# print resultprint("The sorted and flattened list : " + str(res)) |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list : [1, 3, 3, 5, 7, 9, 12]
The time complexity of the provided code is O(nlogn), where n is the total number of elements in the input list of lists.
The auxiliary space complexity of the code is O(n), where n is the total number of elements in the input list of lists.
Method #3: Using the sorted function and the sum function:
Here is another approach using the sorted function and the sum function:
- Python3
Python3
# Python3 code to demonstrate# sort flatten list of list # initializing list of list test_list = [[3, 5], [7, 3, 9], [1, 12]]# Use the sum function to concatenate the lists in test_list and# pass an empty list as the initial value.# Then use the sorted function to sort the resulting list.result = sorted(sum(test_list, []))#Printing resultprint("Original list:", test_list)print("Flattened and sorted list:", result)#This code is contributed by Edula Vinay Kumar Reddy |
Original list: [[3, 5], [7, 3, 9], [1, 12]] Flattened and sorted list: [1, 3, 3, 5, 7, 9, 12]
This code first uses the sum function to concatenate the lists in test_list and pass an empty list as the initial value. This results in a new list that is the concatenation of the lists in test_list. Then, the sorted function is used to sort the resulting list.
In terms of time complexity, this code has a complexity of O(n * log(n)) since it needs to concatenate the lists in test_list and then sort the resulting list, which both have a complexity of O(n). In terms of space complexity, it has a complexity of O(n) since it creates a new list that is the concatenation of the lists in test_list.
Method #4 : Using sort() and extend() methods
- Python3
Python3
# Python3 code to demonstrate# sort flatten list of list# initializing list of listtest_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of listprint("The original list : " + str(test_list))# using sorted + list comprehensionres=[]for i in test_list: res.extend(i)res.sort()# print resultprint("The sorted and flattened list : " + str(res)) |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list : [1, 3, 3, 5, 7, 9, 12]
Time Complexity : O(N*logN)
Auxiliary Space : O(1)
Method #5 : Using a nested for loop:
- Python3
Python3
# Initializing list of list test_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of listprint("The original list : " + str(test_list)) # Flattening the list of lists and storing the result in resres = []for sub_list in test_list: for item in sub_list: res.append(item)# Sorting the res listres.sort()# Printing the sorted res listprint("The sorted and flattened list : " + str(res))#This code is contributed by Jyothi pinjala. |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list : [1, 3, 3, 5, 7, 9, 12]
Time Complexity: O(N log n)
Auxiliary Space: O(N)
Method#6 : Using reduce and operator.add
- Python3
Python3
import functoolsimport operator# initialize the list of liststest_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of listprint("The original list : " + str(test_list))# use functools.reduce and operator.add to flatten the list of listsflattened_list = functools.reduce(operator.add, test_list)# sort the flattened listsorted_list = sorted(flattened_list)# print the resultprint("The sorted and flattened list:", sorted_list)#This code is contributed by Vinay Pinjala. |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list: [1, 3, 3, 5, 7, 9, 12]
Time Complexity: O(N log N)
Auxiliary Space: O(N)
Method#7: Using Recursive method.
- Python3
Python3
# Python3 code to demonstrate# sort flatten list of listdef flatten_and_sort(lst): flat_list = [] for i in lst: if type(i) == list: flat_list.extend(flatten_and_sort(i)) else: flat_list.append(i) return sorted(flat_list)# initializing list of listtest_list = [[3, 5], [7, 3, 9], [1, 12]]# printing original list of listprint("The original list : " + str(test_list))# using flatten_and_sort()res = flatten_and_sort(test_list)# print resultprint("The sorted and flattened list : " + str(res))#this code contributed by tvsk |
The original list : [[3, 5], [7, 3, 9], [1, 12]] The sorted and flattened list : [1, 3, 3, 5, 7, 9, 12]
Time Complexity: O(n)
Auxiliary Space: O(n)
Method # 8 : Using a stack
- Create an empty stack and push the input list onto it.
- Create an empty list to store the flattened list.
- While the stack is not empty, pop the top element from the stack.
- If the popped element is a list, then push its elements onto the stack.
- If the popped element is not a list, then append it to the flattened list.
- Sort the flattened list using the sorted() function.
- Python3
Python3
def flatten_and_sort(lst): stack = [lst] flat_list = [] while stack: element = stack.pop() if isinstance(element, list): stack.extend(element) else: flat_list.append(element) return sorted(flat_list)# Example usagetest_list = [[3, 5], [7, 3, 9], [1, 12]]print("Original list:", test_list)result = flatten_and_sort(test_list)print("Flattened and sorted list:", result) |
Original list: [[3, 5], [7, 3, 9], [1, 12]] Flattened and sorted list: [1, 3, 3, 5, 7, 9, 12]
Time complexity: O(n log n) for the sort operation, where n is the total number of elements in the list of lists.
Auxiliary space: O(n), where n is the total number of elements in the list of lists, for the stack and flattened list.
