Matlab Get Index Of Max Value In Python