如何在pandas中有条件地拆分全名字符串列

import pandas as pd

# 创建包含全名字符串的DataFrame
data = {'FullName': ['John Smith', 'Jane Doe', 'Michael Johnson']}
df = pd.DataFrame(data)

# 使用str.contains()方法检查每个字符串是否包含空格
has_space = df['FullName'].str.contains(' ')

# 根据条件进行拆分，如果包含空格，则按照空格进行拆分
df['FirstName'] = df.loc[has_space, 'FullName'].str.split(' ').str[0]
df['LastName'] = df.loc[has_space, 'FullName'].str.split(' ').str[1]

# 如果不包含空格，则将全名作为FirstName，LastName为空
df.loc[~has_space, 'FirstName'] = df.loc[~has_space, 'FullName']
df.loc[~has_space, 'LastName'] = ''

# 输出结果
print(df)

          FullName  FirstName  LastName
0      John Smith        John     Smith
1        Jane Doe       Jane       Doe
2  Michael Johnson   Michael   Johnson