Numpy Array Indexing Like Matlab Simulink