python - 在 Pandas 数据框中减去 Col A - Col B

Question

我在数据框中有两个字符串列，我想从 A 中减去 B 中的常用词。

A ->  Stack Overlflow is great            


B -> stack great


A-B -> overflow is

我试过下面的代码。但只有当 B 列只有一个词时才有效。

df['A-B'] = [' '.join(set(a.split())-set(b.split())) for a, b in zip(df['A'], df['B'])]

我可以做些什么来让它在 B 有多个单词时起作用吗？

Answer 1

示例 df 是:

>>> df = pd.DataFrame({'A': ['Stack Overlflow is great'], 'B': ['stack great']})

你可以使用apply:

>>> df['A-B'] = df.apply(lambda x: ' '.join([i for i in x[0].split() if i.lower() not in x[1].split()]), axis=1)
>>> df
                          A            B           A-B
0  Stack Overlflow is great  stack great  Overlflow is
>>>