Колонка перенумерации Панды Dataframe

Question

У меня есть простой dataframe как:

>>> df_test = pd.DataFrame(columns=["ID" ]) 
>>> df_test["DATE"] = [ '2001-02-01', '2001-03-01', '2001-02-15', '2001-03-15', '2001-02-15', '2001-03-30'] 
>>> df_test['ID'] = [1,2,3,1,5,4] 
>>> df_test 
    ID  DATE 
0 1 2001-02-01 
1 2 2001-03-01 
2 3 2001-02-15 
3 1 2001-03-15 
4 5 2001-02-15 
5 4 2001-03-30 

Я хотел бы изменить порядок идентификаторов столбцов следующим образом:

>>> df_test 
    ID  DATE 
0 1 2001-02-01 
1 1 2001-03-01 
2 1 2001-02-15 
3 1 2001-03-15 
4 4 2001-02-15 
5 4 2001-03-30 

в основном каждый раз, когда есть ID меньше, чем предыдущая один заполняет предыдущие значения с помощью небольшого идентификатора, чтобы сохранить монотонность. Я пытаюсь со следующим:

df_test.ID.lt(df_test.ID.shift()) 

, что дает мне точку изменения в монотонности:

0 False 
1 False 
2 False 
3  True 
4 False 
5  True 
Name: ID, dtype: bool 

Но я не знаю, как двигаться вперед. Я был бы признателен любой идеей

Спасибо!

Answer 1

Как насчет этого. Не очень элегантный, но, видимо, работает (вам нужно отлаживать это на реальных данных). Обратите внимание: здесь важна петля, потому что каждая итерация должна работать с текущими данными, а не с исходной.

prev_max = 0 
for i in range(1, 6): 
    cur_max = df_test[df_test.ID == i].index.max() 
    if pd.notnull(cur_max): 
     df_test.loc[prev_max:cur_max, 'ID'] = i 
     prev_max = cur_max + 1