如果匹配另一个列表中的值，则提取字符串

Question

我想获取查找列表的值，而不是布尔值。我尝试了以下代码：

val = pd.DataFrame(['An apple','a Banana','a cat','a dog'])
lookup = ['banana','dog']
# I tried the follow code:
val.iloc[:,0].str.lower().str.contains('|'.join(lookup))
# it returns:
0    False
1     True
2    False
3     True
Name: 0, dtype: bool

我想要什么：

0    False
1    banana
2    False
3    dog

感谢您的帮助。

Answer 1

您可以使用 extract 代替 contains ，并且可以将fillna与False一起使用：

import re
p = rf'\b({"|".join(lookup)})\b'
val[0].str.extract(p, expand=False, flags=re.I).fillna(False)

        0
0   False
1  banana
2   False
3     dog