如何在Python中对address列应用拆分字符串操作？

pip install pandas

import pandas as pd

# 假设你有一个DataFrame 'df'，其中包含一个名为'address'的列
data = {'address': ['123 Main St, Cityville', '456 Elm St, Townsville']}
df = pd.DataFrame(data)

# 使用str.split()方法拆分'address'列，默认以空格为分隔符
df[['street', 'city']] = df['address'].str.split(', ', expand=True)

print(df)

              address       street      city
0  123 Main St, Cityville  123 Main St  Cityville
1  456 Elm St, Townsville  456 Elm St  Townsville

# 自定义拆分函数
def split_address(address):
    parts = address.split(', ')
    street_parts = parts[0].split(' ')
    street_number = street_parts[0]
    street_name = ' '.join(street_parts[1:])
    return pd.Series([street_number, street_name, parts[1]])

# 应用自定义拆分函数
df[['street_number', 'street_name', 'city']] = df['address'].apply(split_address)

print(df)

              address street_number street_name      city
0  123 Main St, Cityville          123   Main St  Cityville
1  456 Elm St, Townsville          456     Elm St  Townsville