在Python/pandas中使用regex查找城市名称

import pandas as pd

# 创建一个包含城市名称的DataFrame
data = {'city': ['New York', 'Los Angeles', 'San Francisco', 'Chicago', 'Houston']}
df = pd.DataFrame(data)

# 使用regex查找城市名称
pattern = r'\b[A-Z][a-z]+\b'  # 正则表达式模式，匹配首字母大写的单词
df['city_name'] = df['city'].str.extract(pattern)

# 打印结果
print(df)

            city    city_name
0       New York     New
1    Los Angeles     Los
2  San Francisco     San
3        Chicago  Chicago
4        Houston  Houston