結果 : how to take user input in function in python