Find the percentage | Hackerrank

find_the_percentage.py

import math

if __name__ == '__main__':
    n = int(input())
    student_marks = {}
    for _ in range(n):
        name, *line = input().split()
        scores = list(map(float, line))
        student_marks[name] = scores
    query_name = input()
    marks_list = student_marks.get(query_name)
    a = sum(marks_list)/len(marks_list)
    print("%.2f" % a)