Python Numpy Array Of Size N