Заменить значение в столбце на основе совпадения, полученного с помощью регулярного выражения, в другом столбце (Python Pandas)

Question

У меня есть кадр данных pandas вроде этого:

       AGENCY| CLIENT
     ___________________________
     AAA     | CLIENT_ONE
     AAB     | CLIENT_TWO SOMETHING
     AAF     | CLIENT_TWO SOMETHING and 55
     KAK     | CLIENT THREE

Если столбец CLIENT содержит строку 'CLIENT_TWO' (за которой следует что-то или нет), я хочу заменить значение соответствующего столбец AGENCY со строкой «HELLO». Окончательный результат должен выглядеть так:

       AGENCY| CLIENT
     ___________________________
     AAA     | CLIENT_ONE
     HELLO   | CLIENT_TWO SOMETHING
     HELLO   | CLIENT_TWO SOMETHING else 55
     KAK     | CLIENT THREE

Как это сделать с помощью регулярного выражения?

Answer 1

Вы можете использовать str.contains и loc:

df.loc[df['CLIENT'].str.contains('CLIENT_TWO'), 'AGENCY'] = 'HELLO'