在数据框的一列的每一行中使用拆分函数

Question

我有一个相当大的 pandas 数据框（超过 100 万行），其中的列包含字符串或数字。现在我想在“应用”表达式之前将字符串拆分为一列。

一个例子来解释我的意思：

我有什么：

a    b    description
2    4    method A is applied
10   5    titration is applied
3    1    computation is applied

我在寻找什么：

 a    b    description
 2    4    method A 
 10   5    titration 
 3    1    computation 

我尝试了以下，

df.description = df.description.str.split('is applied')[0]

但这并没有带来预期的结果。

任何想法如何做到这一点？ :-)

Answer 1

你很亲密，需要str[0]：

df.description = df.description.str.split(' is applied').str[0]

替代解决方案：

df.description = df.description.str.extract('(.*)\s+is applied')

---

print (df)
    a  b  description
0   2  4     method A
1  10  5    titration
2   3  1  computation

但为了获得更好的性能，请使用list comprehension:

df.description = [x.split(' is applied')[0] for x in df.description]

Answer 2

你可以使用replace

df.description = df.description.str.replace(' is applied','')
df
    a  b  description
0   2  4     method A
1  10  5    titration
2   3  1  computation