結果 : how to take input in function in python