結果 : how to define an array of size n in python