Как установить значение пустых столбцов на основе некоторых условий

#python #pandas

Вопрос:

Я использую панд. У меня есть следующая проблема:

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

Я хочу установить столбцы B как значение столбца A, для которого C имеет значение NA, до тех пор, пока не появится следующее значение NA в столбце C. Вот ожидаемый результат:

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

Комментарии:

1. df['B'] = df.assign(B=np.where(df['C'].isna(), df['A'], np.nan))['B'].ffill() ?

2. Спасибо, это работает