Python Create Array With N Elements