Create Array With N Elements Python