Python Define Numpy Array