python dataframe 模糊匹配_Python模糊匹配(FuzzyWuzzy) – 保持最佳匹配

我正在尝试模糊匹配两个csv文件,每个文件包含一列相似但不相同的名称.

我的代码到目前为止如下：

import pandas as pd

from pandas import DataFrame

from fuzzywuzzy import process

import csv

save_file = open('fuzzy_match_results.csv', 'w')

writer = csv.writer(save_file, lineterminator = '\n')

def parse_csv(path):

with open(path,'r') as f:

reader = csv.reader(f, delimiter=',')

for row in reader:

yield row

if __name__ == "__main__":

## Create lookup dictionary by parsing the products csv

data = {}

for row in parse_csv('names_1.csv'):

data[row[0]] = row[0]

## For each row in the lookup compute the partial ratio

for row in parse_csv("names_2.csv"):

#print(process.extract(row,data, limit = 100))

for found, score, matchrow in process.extract(row, data, limit=100):

if score >= 60:

print('%d%% partial match: "%s" with "%s" ' % (score, row, found))

Digi_Results = [row, score, found]

writer.writerow(Digi_Results)

save_file.close()

输出如下：

Name11 , 90 , Name25

Name11 , 85 , Name24

Name11 , 65 , Name29

该脚本工作正常.输出符合预期.

但我正在寻找的只是最好的匹配.

Name11 , 90 , Name25

Name12 , 95 , Name21

Name13 , 98 , Name22

因此,我需要根据第2列中的最高值以某种方式删除第1列中的重复名称.

它应该是相当简单的,但我似乎无法弄明白.

任何帮助,将不胜感激.