Python | Get Top N elements from Records
Sometimes, while working with data, we can have a problem in which we have records and we require to find the highest N scores from it. This kind of application is popular in web development domain. Let’s discuss certain ways in which this problem can be solved.
Method #1 : Using sorted() + lambda
The combination of above functionality can be used to perform this particular task. In this, we just employ sorted function with reverse flag true, and print the top N elements using list slicing.
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
# Python3 code to demonstrate working of# Get Top N elements from Records# Using sorted() + lambda # Initializing list test_list = [('Manjeet', 10), ('Akshat', 4), ('Akash', 2), ('Nikhil', 8)] # Initializing N N = 2 # printing original listprint("The original list is : " + str(test_list)) # Get Top N elements from Records# Using sorted() + lambdares = sorted(test_list, key = lambda x: x[1], reverse = True)[:N] # printing resultprint("The top N records are : " + str(res)) |
The original list is : [('Manjeet', 10), ('Akshat', 4), ('Akash', 2), ('Nikhil', 8)]
The top N records are : [('Manjeet', 10), ('Nikhil', 8)]
Method #2 : Using sorted() + itemgetter()
The combination of above functions can also be used to perform this particular task. In this, the task performed by lambda function is performed by itemgetter() is used to get the index in tuple which has to be included in calculations.
# Python3 code to demonstrate working of# Get Top N elements from Records# Using sorted() + itemgetter()from operator import itemgetter # Initializing list test_list = [('Manjeet', 10), ('Akshat', 4), ('Akash', 2), ('Nikhil', 8)] # Initializing N N = 2 # printing original listprint("The original list is : " + str(test_list)) # Get Top N elements from Records# Using sorted() + itemgetter()res = sorted(test_list, key = itemgetter(1), reverse = True)[:N] # printing resultprint("The top N records are : " + str(res)) |
The original list is : [('Manjeet', 10), ('Akshat', 4), ('Akash', 2), ('Nikhil', 8)]
The top N records are : [('Manjeet', 10), ('Nikhil', 8)]
