Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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())
Advertisement
Add Comment
Please, Sign In to add comment