Как отсортировать эффективно по второму аргументу?

Question

[arklight]

Цель: написать программу, которая будет сортировать людей по баллам за 3 предмета
На входе: первая строка - фамилия, вторая строка - баллы за 3 предмета через пробел
Исходный код

data = []

def insert_data():
    str = input().split(" ")
    name = str[0]
    balli = int(str[1]) + int(str[2]) + int(str[3])
    data.append([name, balli])


for i in range(int(input())):
    insert_data()

data.sort(reverse=True, key=lambda item: item[1])
for item in data:
    print(*item)

Как можно после сортировки по баллам, эффективно по затратам времени, отсортировать по фамилиям так, чтобы сортировка по баллам не нарушилась? То есть, если есть 2 человека Афанасьев 200 и Арефьев 200, вначале чтобы стоял Арефьев, а потом Афаньсьев

Answer 1

[o5a]

Чтобы сортировать сразу по нескольким колонкам, можно использовать соответствующие наборы элементов в кортеже/списке, т.е. изменить:
data.sort(key=lambda item: (-item[1], item[0]) )

Comments

[arklight]

Это не эффективно по времени и про этот способ я знаю

[o5a]

Максим, вообще можно делать последовательные сортировки, в обратном порядке приоритетов, т.е. сначала по фамилиям, затем по оценкам, последующая сортировка не нарушает порядок предыдущей без необходимости

data.sort(key=lambda item: item[0])
data.sort(reverse=True, key=lambda item: item[1])

[arklight]

o5a, нет, не принимает так, time limit

[o5a]

Максим, Что-то мне кажется, у тебя проблема не в выбранном способе сортировки, а в реализации ввода/вывода (или еще чего побочного, что здесь не упомянул). Сомнительно, что подобная сортировка стала узким местом.

Например, представленный код ввода данных расходится с описанием.

Возможно, time limit происходит из-за использования input() как раз, или еще чего, не упомянутого тут.

[arklight]

o5a, да, может быть из-за реализации, но там строгие правила. Первая строка - кол-во элементов, вторая строка и далее - сами элементы. Если есть предположения, как это ускорить - буду рад прочитать

[o5a]

Максим, а с какого сайта вообще задача, можно ссылку на задачу глянуть? Возможно просто решение ожидает обработку уже переданных данных, а не использование input()

[arklight]

o5a, да, конечно, вот vtcloud5.ulstu.ru/ru/problem-pid-c531?ps=1&smt=a&s... . Но там надо именно чтобы ввод был

Answer 2

[aRegius]

Попробуйте в качестве аргумента параметра key использовать функцию itemgetter модуля operator:

from operator import itemgetter
data.sort(key=itemgetter(1, 0))

Comments

[arklight]

Я, конечно, попробую завтра, но что-то мне подсказывает, что это будет такая же двойная сортировка, а как и key lambda, которая выше (или частично в моём примере). Обычная сортировка с кортежем приводит к time limit