building up an array in numpy/scipy by iteration in python?,"array([0, 1, 4, 9, 16])"