結果 : function to take input in python