Python Creating Array Of Size N