Matlab Get Index Of Max Value In Array In Python