Как вычислить квадрат суммы в языке Python