Python Create Array Of Size N With Value 1