Create Array Of Specific Size Python