Guest User

Untitled

a guest
Apr 20th, 2018
244
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.37 KB | None | 0 0
  1. import sys
  2. import difflib
  3. import csv
  4. import pandas as pd
  5.  
  6. df = pd.read_csv(sys.argv[1])
  7. List = df.list #.astype(str).values.tolist()
  8. MappedID = df.Party.astype(str)
  9. df['Name_r'] = MappedID.map(lambda x: (difflib.get_close_matches(x, List, cutoff=0.7)[:1] or [None][0]))
  10. header = [ "List", "Party", "Name_r"]
  11. df.to_csv(sys.argv[2], columns = header)
  12. print(df.to_string())
Advertisement
Add Comment
Please, Sign In to add comment