Expected output: 56.00 and my output: 56.0
from statistics import mean
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=list(student_marks[query_name])
    ans=(mean(marks_list))
    print(round(ans),2)
 
My answer is 56.0 for the input, so is there any short way to do this?
 
     
     
     
    