Find Index Array Python