結果 : how to return an array in python function