結果 : how to ask for user input in a function python