Initialize Array Of Certain Size Python