結果 : how to get user input in a function in python