Python Create Array Of Size N With Line