python - 在 pandas DataFrame 中快速应用字符串操作

Question

假设我有一个 DataFrame，它有 10 万行和一列 name。我想尽可能有效地将这个名字分成名字和姓氏。我目前的方法是，

def splitName(name):
  return pandas.Series(name.split()[0:2])

df[['first', 'last']] = df.apply(lambda x: splitName(x['name']), axis=1)

不幸的是，DataFrame.apply 真的非常慢。我能做些什么来使这个字符串操作几乎和 numpy 操作一样快吗？

谢谢!

Answer 1

尝试(需要 pandas >= 0.8.1):

splits = x['name'].split()
df['first'] = splits.str[0]
df['last'] = splits.str[1]