結果 : how to get complex number as input in python