Вопрос по информатике:

Дан одномерный массив. Найти среднее арифметическое его элементов. Вывести на экран только те элементы массива, которые больше найденного среднего арифметического.
НА ПИТОНЕ!!!!!!!!!!!!
ХЕЛППППППППППППП!!!!!!!!!!!

array = []

for i in range(int(input())):

   array.append(int(input()))

average = sum(array)/len(array)

print(*[i for i in array if i > average])

P.S. Первые 3 строки - создание массива, 2 последние - решение

P.S.S. Если что-то не так, то скажи, исправлю.