import sys import difflib import csv import pandas as pd df = pd.read_csv(sys.argv[1]) List = df.list #.astype(str).values.tolist() MappedID = df.Party.astype(str) df['Name_r'] = MappedID.map(lambda x: (difflib.get_close_matches(x, List, cutoff=0.7)[:1] or [None][0])) header = [ "List", "Party", "Name_r"] df.to_csv(sys.argv[2], columns = header) print(df.to_string())