結果 : user input in a function python