Python Numpy Index Array Matlab