通过匹配python中的其他列值来拆分列值

import pandas as pd

# 创建一个示例数据集
data = {
    'Name': ['John', 'Alice', 'Bob'],
    'Age': [25, 30, 35],
    'City': ['New York', 'Paris', 'London'],
    'Country': ['USA', 'France', 'UK']
}

df = pd.DataFrame(data)

# 定义一个函数来根据其他列的值拆分列的值
def split_column(row):
    if row['Country'] == 'USA':
        return row['City'].split(' ')[0]  # 以空格拆分并返回第一个部分
    else:
        return row['City']

# 应用函数到要拆分的列
df['Split_City'] = df.apply(split_column, axis=1)

# 打印结果
print(df)