結果 : how to make an array of length n in python