結果 : define function python with input