High Five Python Solution with Tests

lcpy gen -n 1086   # by problem number
lcpy gen -s high_five   # by problem name

Input: items = [[1,91],[1,92],[2,93],[2,97],[1,60],[2,77],[1,65],[1,87],[1,100],[2,100],[2,76]]
Output: [[1,87],[2,88]]
Explanation:
The student with ID = 1 got scores 91, 92, 60, 65, 87, and 100. Their top five average is (100 + 92 + 91 + 87 + 65) / 5 = 87.
The student with ID = 2 got scores 93, 97, 77, 100, and 76. Their top five average is (100 + 97 + 93 + 77 + 76) / 5 = 88.6, but with integer division their average converts to 88.

Input: items = [[1,100],[7,100],[1,100],[7,100],[1,100],[7,100],[1,100],[7,100],[1,100],[7,100]]
Output: [[1,100],[7,100]]

from collections import defaultdict
from heapq import nlargest


class Solution:
    # Time: O(n log n) for sorting scores
    # Space: O(n)
    def high_five(self, items: list[list[int]]) -> list[list[int]]:
        scores: dict[int, list[int]] = defaultdict(list)
        for student, score in items:
            scores[student].append(score)
        return [[student, sum(nlargest(5, vals)) // 5] for student, vals in sorted(scores.items())]