Create An Array With N Elements Python