Posted By

eristoddle on 06/30/11


Tagged

python csv


Versions (?)

Python CSV Row to Object


 / Published in: Python
 

  1. import csv
  2.  
  3. def fix_name(n):
  4. 'Fixes a string to be nicer (usable) variable names.'
  5. return n.replace(' ','_').replace('.','_')
  6.  
  7. def read_csv(inf, close_file=True):
  8. 'Reads a csv file in as a list of objects'
  9. def from_csv_line(l, h):
  10. return DictObj(dict(zip(h, l)))
  11. iter = csv.reader(inf).__iter__()
  12. header = map(fix_name, iter.next())
  13. for i in iter:
  14. yield from_csv_line(i, header)
  15. if close_file:
  16. inf.close()
  17.  
  18. def write_csv(out, objs, cols=None, close_file=True):
  19. 'Writes a list of dicts out as a csv file.'
  20. def to_csv_line(obj, header):
  21. def lookup_or_empty(o, k, default=''):
  22. if k in o:
  23. return o[k]
  24. return default
  25. return map(lambda h : lookup_or_empty(obj,h), header)
  26. iter = objs.__iter__()
  27. output = csv.writer(out)
  28. first = iter.next()
  29. if cols is None:
  30. cols = sorted(first.keys())
  31. output.writerow(cols)
  32. output.writerow(to_csv_line(first, cols))
  33. output.writerows(map(lambda x: to_csv_line(x, cols), iter))
  34. if close_file:
  35. out.close()

Report this snippet  

You need to login to post a comment.