Python program to Sort Matrix by Maximum Row element

Given a Matrix, sort rows by maximum element.

Input : test_list = [[5, 7, 8], [9, 10, 3], 
[10, 18, 3], [0, 3, 5]] 
Output : [[10, 18, 3], [9, 10, 3], [5, 7, 8], [0, 3, 5]] 
Explanation : 18, 10, 8 and 5 are maximum elements in rows, hence sorted.

Input : test_list = [[9, 10, 3], 
[10, 18, 3], [0, 3, 5]] 
Output : [[10, 18, 3], [9, 10, 3], [0, 3, 5]] 
Explanation : 18, 10, and 5 are maximum elements in rows, hence sorted. 

Method #1 : Using sort() + max()

In this, we perform task of sorting using sort() with key being maximum element from each row. The reverse keyword is used to sort by keeping maximum element rows at start and decreasing from there.

# Python3 code to demonstrate working of 
# Sort Matrix by Maximum Row element
# Using sort() + max()

def max_sort(row):
    return max(row)

# initializing list
test_list = [[5, 7, 8], [9, 10, 3],
             [10, 18, 3], [0, 3, 5]]

# printing original list
print("The original list is : " + str(test_list))

# sort() for sorting, max to get maximum values
test_list.sort(key = max_sort, reverse = True)

# printing result 
print("The maximum sorted Matrix : " + str(test_list))

Output:

The original list is : [[5, 7, 8], [9, 10, 3], [10, 18, 3], [0, 3, 5]] The maximum sorted Matrix : [[10, 18, 3], [9, 10, 3], [5, 7, 8], [0, 3, 5]]

Time Complexity: O(nlogn) where n is the number of elements in the list “test_list”. 
Auxiliary Space: O(n) additional space of size n is created where n is the number of elements in the list “test_list”. 

Method #2 : Using sorted() + lambda + max()

In this, we perform task of sorting using sorted() for non-inplace sort, and lambda function is used instead of external function to include maximum element from row logic.

# Python3 code to demonstrate working of 
# Sort Matrix by Maximum Row element
# Using sorted() + lambda + max()

# initializing list
test_list = [[5, 7, 8], [9, 10, 3],
             [10, 18, 3], [0, 3, 5]]

# printing original list
print("The original list is : " + str(test_list))

# sorted() for sorting, max to get maximum values
# reverse for reversed order
res = sorted(test_list, key = lambda row : max(row), reverse=True)

# printing result 
print("The maximum sorted Matrix : " + str(res))

Output:

The original list is : [[5, 7, 8], [9, 10, 3], [10, 18, 3], [0, 3, 5]] The maximum sorted Matrix : [[10, 18, 3], [9, 10, 3], [5, 7, 8], [0, 3, 5]]

Method 3: Using a for loop

Step-by-step approach:

• Initialize the given matrix test_list.
• Iterate through each row of the matrix using a for loop.
• For each row, find the maximum element using the max() function and store it in a list max_list.
• Use the zip() function to pair the elements of max_list and test_list row-wise.
• Sort the paired elements based on the values of max_list in descending order using the sorted() function and list comprehension.
• Unzip the sorted pairs using the zip() function and store only the second element of each pair (i.e. the sorted rows of the matrix) in a list res.
• Print the sorted matrix res.

# Python3 code to demonstrate working of 
# Sort Matrix by Maximum Row element
# Using for loop + max() + sort()

# initializing list
test_list = [[5, 7, 8], [9, 10, 3],
             [10, 18, 3], [0, 3, 5]]

# printing original list
print("The original list is : " + str(test_list))

# for loop to iterate through each row and store maximum element in a list
max_list = []
for row in test_list:
    max_list.append(max(row))

# sort matrix based on maximum values using zip()
res = [x for _, x in sorted(zip(max_list, test_list), reverse=True)]

# printing result 
print("The maximum sorted Matrix : " + str(res))

The original list is : [[5, 7, 8], [9, 10, 3], [10, 18, 3], [0, 3, 5]]
The maximum sorted Matrix : [[10, 18, 3], [9, 10, 3], [5, 7, 8], [0, 3, 5]]

Time complexity: O(n^2) as we need to iterate through each element of the matrix and find the maximum element of each row.
Auxiliary space: O(n) to store the maximum values of each row in a list.

manjeet_04

Follow

Improve

Article Tags :

• Python
• Python Programs
• Python list-programs

Practice Tags :

• python