Guest User

Untitled

a guest
Feb 12th, 2013
140
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.98 KB | None | 0 0
  1. pd.read_fwf(path, colspecs=markers, names=columns,
  2. converters=create_convert_dict(columns))
  3.  
  4. def create_convert_dict(columns):
  5. convert_dict = {}
  6. for col in columns:
  7. convert_dict[col] = null_convert
  8. return convert_dict
  9.  
  10. def null_convert(value):
  11. value = value.strip()
  12. if value == "":
  13. return None
  14. else:
  15. return value
  16.  
  17. pd.read_fwf(path, colspecs=markers, names=columns, na_values='',
  18. converters=create_convert_dict(columns))
  19.  
  20. def create_convert_dict(columns):
  21. convert_dict = {}
  22. for col in columns:
  23. convert_dict[col] = col_strip
  24. return convert_dict
  25.  
  26. def col_strip(value):
  27. return value.strip()
  28.  
  29. In [57]: data = """
  30. A B C
  31. 0 foo
  32. 3 bar 2.0
  33. 1 3.0
  34. """
  35.  
  36. In [58]: df = pandas.read_fwf(StringIO(data), widths=[5, 5, 5])
  37.  
  38. In [59]: df
  39. Out[59]:
  40. A B C
  41. 0 0 foo NaN
  42. 1 3 bar 2
  43. 2 1 NaN 3
  44.  
  45. In [60]: df.dtypes
  46. Out[60]:
  47. A int64
  48. B object
  49. C float64
Advertisement
Add Comment
Please, Sign In to add comment