HackerRank Python Solution - Built-Ins - Athlete Sort

• You are given a spreadsheet that contains a list of N athletes and their details (such as age, height, weight, and so on). You are required to sort the data based on the K^th attribute and print the final resulting table. Follow the example given below for a better understanding.

• Note that K is indexed from 0 to M-1, where M is the number of attributes.
• Note: If two attributes are the same for different rows, for example, if two athletes are of the same age, print the row that appeared first in the input.

Input Format:

• The first line contains N and M separated by a space.
• The next N lines each contain M elements.
• The last line contains K.

Constraints:

• 1 <= N, M <= 1000
• 0 <= K < M
• Each Element <= 1000

Output Format:

• Print the N lines of the sorted table. Each line should contain space-separated elements. Check the sample below for clarity.

Sample Input:

5 3
10 2 5
7 1 0
9 9 9
1 23 12
6 5 9
1

Sample Output:

7 1 0
10 2 5
6 5 9
9 9 9
1 23 12

Explanation:

• The details are sorted based on the second attribute since K is zero-indexed.

Solution:

N, M = list(map(int,input().split()))

arr = [input() for _ in range(N)]

k = int(input())

for row in sorted(arr, key=lambda arr: int(arr.split()[k])):
    print(row)

Disclaimer: The problem statement is given by hackerrank.com but the solution is generated by the Geek4Tutorial admin. If there is any concern regarding this post or website, please contact us using the contact form. Thank you!