python - 从 pandas 数据框列值中删除句子的第一个单词

Question

我有一个这样的数据框:

df:
col1      col2
 A        blue berry
 B        nice water bottle

我想从 col2 值中删除第一个单词，最终数据框将如下所示:

df1:
col1       col2
 A         berry
 B         water bottle

如何使用 pandas 以最有效的方式做到这一点

Answer 1

使用split通过第一个带有 n=1 的空格，然后通过索引选择第二个列表:

df['col2'] = df['col2'].str.split(n=1).str[1]
print (df)
  col1          col2
0    A         berry
1    B  water bottle

详细信息:

print (df['col2'].str.split(n=1))
0           [blue, berry]
1    [nice, water bottle]
Name: col2, dtype: object

如果性能很重要并且没有缺失值将解决方案转换为列表理解:

df['col2'] = [x.split(maxsplit=1)[1] for x in df['col2']]