We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Built-Ins
  4. Athlete Sort
  5. Discussions

Athlete Sort

  • kumarmayashanka1
     + 0 comments
    #!/bin/python3

import math
import os
import random
import re
import sys

if __name__ == '__main__':
    # Read the first line: number of rows (n) and number of columns (m)
    nm = input().split()

    n = int(nm[0])  # Number of athletes
    m = int(nm[1])  # Number of attributes per athlete

    arr = []

    # Read the next n lines containing athlete details
    for _ in range(n):
        arr.append(list(map(int, input().rstrip().split())))

    # Read the sorting column (k), indexed from 0
    k = int(input())

    # Sort the array based on the k-th attribute using a lambda function
    arr.sort(key=lambda x: x[k])

    # Print the sorted array
    for row in arr:
        print(" ".join(map(str, row)))