Как создать функцию для замены слов

Question

У меня есть df с двумя столбцами: ORIGIN, DESTINATION

Я хочу сделать замену.

Один из них содержит 'string' и заменяет

# assume string contains 'UK|Heathrow' is 'London Heathrow Airport'

dis_air.loc[dis_air['ORIGIN'].str.contains('UK|Heathrow'), 'ORIGIN'] = 'London Heathrow Airport'
dis_air.loc[dis_air['DESTINATION'].str.contains('UK|Heathrow'), 'DESTINATION'] = 'London Heathrow Airport'


# assume string contains 'Gatwick' is 'London Gatwick Airport'
dis_air.loc[dis_air['ORIGIN'].str.contains('Gatwick'), 'ORIGIN'] = 'London Gatwick Airport'
dis_air.loc[dis_air['DESTINATION'].str.contains('Gatwick'), 'DESTINATION'] = 'London Gatwick Airport'

Один равен 'string' и заменяет

# assume 'London' is 'Heathrow'
dis_air.ORIGIN_new [dis_air.ORIGIN_new == 'London'] = 'London Heathrow Airport'
dis_air.DESTINATION_new [dis_air.DESTINATION_new == 'London'] = 'London Heathrow Airport'

Может ли кто-нибудь помочь создать две функции, тогда я могу просто ввести a и b два параметра для достижения результата. a - строка должна быть заменена, b - альтернативная строка.

Answer 1

В pandas уже есть функция для этого.

dis_air['ORIGIN'].replace('UK|Heathrow','London Heathrow Airport',inplace=True)