Advertisement
Guest User

Untitled

a guest
Feb 19th, 2019
76
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.41 KB | None | 0 0
  1. def add_hierarchical_feats(transactions,transaction_type='historical'):
  2.     data = transactions
  3.     aggs = ['mean','min','max','var']
  4.     pairs = [['state_id','city_id'],['subsector_id','merchant_category_id']]
  5.     for agg in aggs:
  6.         for pair in pairs:
  7.             data['{}_{}_{}_{}'.format(transaction_type,pair[0],pair[1],agg)] = data.groupby(pair)['purchase_amount'].transform(agg).fillna(0)
  8.     return data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement