Python Get Index Of For Loop