python - Python - 删除所有不在列表中的子字符串

Question

我想删除定义列表中不存在的 df 列中的所有子字符串。例如：

mylist = {good, like, bad, hated, terrible, liked}

Current:                                         Desired:
index      content                               index        content                                          
0          a very good idea, I like it           0            good like
1          was the bad thing to do               1            bad
2          I hated it, it was terrible           2            hated terrible
...                                              ...
k          Why do you think she liked it         k            liked

我已经设法定义了一个函数，它使所有单词不在列表中，但是不知道如何反转这个函数来实现我想要的：

pat = r'\b(?:{})\b'.format('|'.join(mylist))
df['column1'] = df['column1'].str.contains(pat, '')

任何帮助，将不胜感激。

标签： pythonstringpandas

Answer 1

str.findall与 一起使用str.join：

df['column1'] = df['content'].str.findall('(' + pat + ')').str.join(' ')
print (df)
                         content         column1
0    a very good idea, I like it       good like
1        was the bad thing to do             bad
2    I hated it, it was terrible  hated terrible
3  Why do you think she liked it           liked

或者使用拆分、过滤和连接列出理解：

df['column1'] = df['content'].apply(lambda x: ' '.join([y for y in x.split() if y in mylist]))
print (df)
                         content         column1
0    a very good idea, I like it       good like
1        was the bad thing to do             bad
2    I hated it, it was terrible  hated terrible
3  Why do you think she liked it           liked