Фрейм данных назначает общие критерии, если они не совпадают

Question

У меня есть датафрейм, организованный следующим образом для вагонов. Я хотел бы посчитать ['Железная дорога'], но только если это соответствует 'VER'. Если нет, я хочу, чтобы 'Railroad' переназначил значение на 'NOT' и посчитал этим.

Иерархия данных:

df1 = df.reset_index().groupby(['Homebase','FINAL ETA','Code Description','L_E', 'Railroad'])['Code Description'].size()

Пример вывода:

Homebase    FINAL ETA   Code Description        L_E  Railroad   
Rail2       2018-12-06  Arrival in yard         L    VER          1
            2019-01-04  Arrival in yard         L    VER         10
            2019-01-08  Arrival in yard         L    FIL         16
            2019-01-09  Arrival in yard         L    FIL          5
            2019-01-13  Arrival in yard         L    PAS          1

Желаемый вывод:

Homebase    FINAL ETA   Code Description        L_E  Railroad
Rail2       2018-12-06  Arrival in yard         L    VER          1
            2019-01-04  Arrival in yard         L    VER         10
            2019-01-08  Arrival in yard         L    NOT         16
            2019-01-09  Arrival in yard         L    NOT          5
            2019-01-13  Arrival in yard         L    NOT          1

Answer 1

Похоже, что меняется только столбец железной дороги, попробуйте это:

ver = (df1['Railroad'] == 'VER')
df1['Railroad'] = 'NOT'
df1.loc[ver, 'Railroad'] = 'VER'