将功能应用于多列文本清洗

时间:2019-09-09 16:58:15

标签: python pandas python-applymap

我在数据框中有三列文本,我想对其应用相同的功能。这是我在下面尝试过的。我应该将什么作为参数传递给函数?

def clean_columns():
     df['column'] = df['column'].str.replace('[^\w\s]',' ')\
                  .str.replace('hello',' ')\
                  .str.replace('goodbye',' ')\
                  .str.lower()\
                  .str.split()
df[['Col1', 'Col2', 'Col3']].applymap(clean_columns)  

我不确定如何将函数分别写入每一列并将函数应用于该函数。有什么想法吗?

1 个答案:

答案 0 :(得分:1)

将函数重写为

def clean_columns(col):
    return col.str.replace('[^\w\s]',' ')\
                  .str.replace('hello',' ')\
                  .str.replace('goodbye',' ')\
                  .str.lower()\
                  .str.split()

并仅使用套用:

df[['col1', 'col2', 'col3']] = df[['col1', 'col2', 'col3']].apply(clean_column)