Дан одномерный массив. найти среднее арифметическое его элементов. вывести на экран только те элементы массива, которые больше найденного среднего арифметического. на !

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. Если что-то не так, то скажи, исправлю.