結果 : how to make an array of a certain size python