Даны три вещественных числа. Возвести в квадрат и найти те из них,значения которых неотрицательны.Язык Puthon

#Использовал списки

a = [int(i) for i in input().split()]

b = []

for i in a:

if i >=0:

 b.append(i*i)

else:

 b.append(i)

print('До {}'.format(a))

print('После {}'.format(b))