Count Elements Greater Than Average

1# Take array
2n = int(input("Enter array size: "))
3arr = []
4for i in range(n):
5    arr.append(int(input(f"Element {i+1}: ")))
6
7# Calculate average
8average = sum(arr) / len(arr)
9
10# Count greater than average
11count = 0
12for element in arr:
13    if element > average:
14        count += 1
15
16print(f"Elements greater than average ({average}): {count}")