Magnitude Of A Complex Number Python