Given a Matrix, the following article extracts a specifies number of rows that has a maximum sum.
Input : test_list = [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]], K = 3
Output : [[199], [2, 3, 4, 5, 6], [3, 4, 5, 6]]
Explanation : 199 > 20 > 18, 3 maximum elements rows are extracted.
Input : test_list = [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]], K = 2
Output : [[199], [2, 3, 4, 5, 6]]
Explanation : 199 > 20, 2 maximum elements rows are extracted.
Method 1 : Using sorted(), reverse, slice and sum()
In this, we perform the task of sorting using sorted() and getting sum using sum(). Reverse key is used to perform reversal of rows for getting maximum summation rows at top and then top K(specific number of rows) rows are sliced.
Python3
# initializing listtest_list = [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]] # printing original listprint("The original list is : " + str(test_list)) # initializing KK = 3 # sorted gets reverse sorted matrix by sum# K rows extracted using slicingres = sorted(test_list, key=lambda row: sum(row), reverse=True)[:K] # printing resultprint("The filtered rows : " + str(res)) |
Output:
The original list is : [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]]
The filtered rows : [[199], [2, 3, 4, 5, 6], [3, 4, 5, 6]]
Method 2 : Using sort(), reverse, slicing and sum()
In this, we perform the task of in place sorting using sort(), using reverse as key. Slicing is done using slice operation. The sum(), is used to take summation and an external function is called to compute sum of rows of the list.
Python3
# row sum util.def row_sum(row): return sum(row) # initializing listtest_list = [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]] # printing original listprint("The original list is : " + str(test_list)) # initializing KK = 3 # sort() used to sort# K rows extracted using slicingtest_list.sort(key=row_sum, reverse=True)res = test_list[:K] # printing resultprint("The filtered rows : " + str(res)) |
Output:
The original list is : [[3, 4, 5, 6], [1, 4, 6], [199], [2, 3, 4, 5, 6], [7, 3, 1]]
The filtered rows : [[199], [2, 3, 4, 5, 6], [3, 4, 5, 6]]
Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.
To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course
