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