Numpy Define Array Dimensions Python