将包含连字符的几列数据分隔开，从而删除Python中的元素

Question

我有一个数据集df，我想在Python中分离字符串。

数据

Type                 Id        
aa - generation      aa - generation01 
aa_led - generation  aa_led - generation01
ss - generation      ss- generation01  

所需

Type    Id
aa      aa01
aa_led  aa_led01
ss      ss01

进行

我正试图将这段代码合并到我的脚本中，但是，这段代码用连字符分隔，但我的列不保留名称。

new = wordstring.strip('-').split('-')

欢迎提出任何建议谢谢

Answer 1

如果您只想从df中的每个值中删除generation。您可以使用applymap:

df = df.applymap(lambda x : x.replace('- generation', '').replace(' ',''))

输出：

Type        Id        
0      aa       aa01
1  aa_led   aa_led01
2      ss       ss01