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?