питон, не понимаю, как откорректировать прогу, нужно именно генератором

Программа:

my_list = list(map(int, input().split()))

max_value = max(my_list)

[print(i, end=' ') for i in range(len(my_list)) if my_list[i] == max_value]

a = list(map(int, input().split()))

max_value = max(a)

indices = [i for i in range(len(a)) if a[i] == max_value]

print(*indices)