Advertisement
Guest User

Untitled

a guest
Mar 18th, 2019
72
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.06 KB | None | 0 0
  1. trip.start_date=pd.to_datetime(trip.start_date,infer_datetime_format=True)
  2. trip.end_date=pd.to_datetime(trip.end_date,infer_datetime_format=True)
  3.  
  4. import re
  5. import datetime
  6. def add_datepart(df, fldname, drop=True, time=False):
  7. fld = df[fldname]
  8. fld_dtype = fld.dtype
  9. if isinstance(fld_dtype, pd.core.dtypes.dtypes.DatetimeTZDtype):
  10. fld_dtype = np.datetime64
  11.  
  12. if not np.issubdtype(fld_dtype, np.datetime64):
  13. df[fldname] = fld = pd.to_datetime(fld, infer_datetime_format=True)
  14. targ_pre = re.sub('[Dd]ate$', '', fldname)
  15. attr = ['Year', 'Month', 'Week', 'Day', 'Dayofweek', 'Dayofyear',
  16. 'Is_month_end', 'Is_month_start', 'Is_quarter_end', 'Is_quarter_start', 'Is_year_end', 'Is_year_start']
  17. if time: attr = attr + ['Hour', 'Minute', 'Second']
  18. for n in attr: df[targ_pre + n] = getattr(fld.dt, n.lower())
  19. df[targ_pre + 'Elapsed'] = fld.astype(np.int64) // 10 ** 9
  20. if drop: df.drop(fldname, axis=1, inplace=True)
  21.  
  22. add_datepart(trip, 'start_date', drop=False, time=True)
  23. add_datepart(trip, 'end_date', drop=False, time=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement