从不带NA的字符串匹配创建新列

import pandas as pd

# 创建一个示例数据集
data = {'Name': ['John', 'Alice', 'Bob', 'Jane'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 定义匹配规则和对应的新列值
match_rules = {'New York': 'USA',
               'Paris': 'France',
               'London': 'UK'}

# 使用匹配规则创建新列
df['Country'] = df['City'].map(match_rules)

# 打印结果
print(df)

   Name  Age      City Country
0  John   25  New York     USA
1 Alice   30     Paris  France
2   Bob   35    London      UK
3  Jane   40     Tokyo     NaN