Create Array With N Elements Python Programming