Python Find Index Of Max In Array