結果 : how to make a function return an array in python