結果 : how to assign elements to an array in python