Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def add_hierarchical_feats(transactions,transaction_type='historical'):
- data = transactions
- aggs = ['mean','min','max','var']
- pairs = [['state_id','city_id'],['subsector_id','merchant_category_id']]
- for agg in aggs:
- for pair in pairs:
- data['{}_{}_{}_{}'.format(transaction_type,pair[0],pair[1],agg)] = \
- data.groupby(pair)['purchase_amount'].transform(agg).fillna(0)
- return data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement