absolute value of a complex number python