Python Numpy Array Index Of Max In Matlab