Using Python Create Array Of Certain Size