你可以在这里使用像
fuzzywuzzy这样的Python库,它支持这种类型的任务:
from fuzzywuzzy import process
df.assign(Output=[process.extract(i, df['Col-1'], limit=3) for i in df['Col-2']])
使用process方法,我们可以获得字符串相似的分数,然后选择前3个,如果存在3个:
上面代码的输出:
Col-1 Col-2 Output
0 Update have a account [(Account, 90, 1), (AccountDTH, 64, 2), (Update, 40, 0)]
1 Account account summary [(Account, 90, 1), (AccountDTH, 63, 2), (Credit Card, 38, 4)]
2 AccountDTH Cancel [(Balance, 62, 3), (Credit Card, 43, 4), (Update, 33, 0)]
3 Balance Balance Summary [(Balance, 90, 3), (Credit Card, 38, 4), (Update, 30, 0)]
4 Credit Card Update credit card [(Update, 90, 0), (Credit Card, 90, 4), (AccountDTH, 27, 2)]
为了加快这种比较(原则上它使用Python的序列匹配器),我建议安装python-Levenshtein