Numpy Define Array Dimensions Python Interview