python - 通过在一列字符串中找到确切的单词来创建一个新列

Question

如果列表中的任何单词与数据框字符串列完全匹配，我想创建一个包含 1 或 0 的新列。

list_provided=["mul","the"]
#how my dataframe looks
id  text
a    simultaneous there the
b    simultaneous there
c    mul why

预期输出

id  text                     found
a    simultaneous there the   1
b    simultaneous there       0
c    mul why                  1

第二行赋值为 0，因为 “mul”或“the”在字符串列“text”中不完全匹配

代码尝试到现在

#For exact match I am using the below code
data["Found"]=np.where(data["text"].str.contains(r'(?:\s|^)penalidades(?:\s|$)'),1,0)

如何遍历循环以找到所提供单词列表中所有单词的精确匹配？

编辑:如果我按照 Georgey 的建议使用 str.contains(pattern)，则数据 ["Found"] 的所有行都变为 1

data=pd.DataFrame({"id":("a","b","c","d"), "text":("simultaneous there the","simultaneous there","mul why","mul")})
list_of_word=["mul","the"]
pattern = '|'.join(list_of_word)
data["Found"]=np.where(data["text"].str.contains(pattern),1,0)

Output:
id  text                     found
a    simultaneous there the   1
b    simultaneous there       1
c    mul why                  1
d    mul                      1

找到的列第二行这里应该是0

Answer 1

您可以使用 pd.Series.apply 和带有生成器表达式的 sum 来做到这一点:

import pandas as pd

df = pd.DataFrame({'id': ['a', 'b', 'c'],
                   'text': ['simultaneous there the', 'simultaneous there', 'mul why']})

test_set = {'mul', 'the'}

df['found'] = df['text'].apply(lambda x: sum(i in test_set for i in x.split()))

#   id                    text  found
# 0  a  simultaneous there the      1
# 1  b      simultaneous there      0
# 2  c                 mul why      1

---

上面提供了一个计数。如果您只需要 bool 值，请使用 any:

df['found'] = df['text'].apply(lambda x: any(i in test_set for i in x.split()))

对于整数表示，链.astype(int)。