結果 : write input and output function in python