Многократная очистка строк в пандах

Question

Я очищаю столбец в кадре данных, который содержит местоположения, как показано ниже:

New York City
New York, NY
New York USA
Las Vegas, Nevada
Las Vegas, NV, USA
Las Vegas North, America

Как очистить строку, чтобы она возвращала только город, т. Е.

New York
New York
New York
Las Vegas
Las Vegas
Las Vegas

Я пробовал df.replace({'Location' : { 'New York.*' : 'New York', 'Las Vegas.*':'Las Vegas'}}) и несколько других вариантов, но не могу заставить его работать.

Любая помощь, высоко ценится

Answer 1

Кажется, что это работает

df['Location'] = df['Location'].str.replace(r'(^.*New York.*$)', 'New York')
df['Location'] = df['Location'].str.replace(r'(^.*Las Vegas.*$)', 'Las Vegas')

взято из Заменить всю строку, если она содержит подстроку в пандах