Find Index In Array Python