Find the Percentage Hackerrank

findthepercentage.py

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()

    percentage = 0
    sum = 0
    count = 0
    for i in student_marks[query_name]:
        sum += i
        count += 1
 
    percentage = sum / count
    print("{0:.2f}".format(percentage))