pandas Как указать значения по условию для каждой группы датафрейма

Имеется фрейм данных:

data = {'фрукт': ['груша','огурец','вишня', 'абрикос', 'груша','арбуз','груша','банан', 'груша', 'вишня','яблоко', 'груша', 'вишня', 'абрикос', 'груша', 'банан'],
'страна': ['россия','сша', 'россия','россия', 'франция','ЮАР','франция', 'россия', np.nan,'россия','андорра', 'франция', 'португалия', 'россия', np.nan, 'россия'],
'id': ['01','01','01','02','02','03','03','011', '011', '011','011', '6', '6', '5', '5', '5'],
'месяц': ['март','январь','январь','март', 'сентябрь','март','октябрь', 'март', 'ноябрь', 'январь','январь', 'март', 'январь', 'март', 'июль', 'апрель']        
}
dates = pd.DataFrame(data, columns = ['фрукт','страна', 'id', 'месяц'])

введите сюда описание изображения

Нужно получить новый фрейм, в котором для каждой группы id в колонке месяц указано значение месяца, которое следует после значения "март". Если внутри группы нет других значений, но есть только "март" - оставить значения "март" для этой группы.

Как это сделать

Какой результат нужно получить:

введите сюда описание изображения


Ответы (0 шт):