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

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