結果 : how to make an array of size n python