結果 : how to store elements in array in python