結果 : how to take input inside function in python