List Sorting

def solve(students, col):
    students = [s.split(',') for s in students.split(';')]
    col = int(col)
    if col == 1:
        students.sort(key=lambda x: (x[1], x[0]))
    else:
        students.sort(key=lambda x: (int(x[2]), x[0]))
    for s in students:
        print(' '.join(s))