Numpy Define Array Dimensions Python Ide