結果 : how to write a return function in python