在 python 中使用 split() 函数创建新列

Question

我有一个数据框 COL-A 的列：

COL-A
DE 25
DE 240
BA 14
XI 4R 123
DE 5G 4
XI 2445

我想通过选择最后一个按空格分隔的值来创建列 COL-B。

预期输出：

COL-A     COL-B
DE 25        25
DE 240      240
BA 14        14
XI 4R 123   123
DE 5G 4       4
XI 2445    2445

我应该使用apply函数吗？

Answer 1

你可以使用rsplit()+straccessor+astype()方法：

df['COL-B']=df['COL-A'].str.rsplit(n=1).str[1].astype(int)
#OR
#df['COL-B']=df['COL-A'].str.split().str[-1].astype(int)

df的输出：

    COL-A       COL-B
0   DE 25       25
1   DE 240      240
2   BA 14       14
3   XI 4R 123   123
4   DE 5G 4     4
5   XI 2445     2445