将 function 应用到 Pandas 数据框列

Question

我正在尝试将以下 function 应用于我的 pandas 数据框中的列。 该列有这样的条目---> EMS：糖尿病紧急情况

我正在尝试拆分条目并返回原因 - 即糖尿病紧急情况，但在“：”之后没有空格

我尝试了以下方法：

def splitandstrip(column):
    for i in column:
        new = i.split(':')[1]
        new.strip()
    return new
df['response_reason'] = df.title.apply(splitandstrip)

Answer 1

您的 function 无法按预期工作，因为str.strip返回一个新字符串。 原字符串不变。 你可以使用

def splitandstrip(element):
   return element.split(':')[1].strip()

我建议使用Series.str访问器进行矢量化字符串操作：

>>> df = pd.DataFrame(['EMS: Diabetic Emergency', 'ABC: DEF'], columns=['A'])
>>> df
                         A
0  EMS: Diabetic Emergency
1                 ABC: DEF
>>> df['A'].str.split(':').str.get(1).str.strip()
0    Diabetic Emergency
1                   DEF
Name: A, dtype: object