結果 : how to write array python