結果 : function to take user input in python