Python Take Imaginary Part Definition