結果 : how to write return function in python