結果 : how to define a function that takes an argument in python