为每个组创建一个子组

时间:2018-09-24 03:47:30

标签: python

我的数据框如下:

orderid     ordername 
1           a
1           b
1           c
2           d
2           e
3           f

我想为每个订单编号创建一个suborder_id列。

suborder_id
   1
   2
   3
   1
   2
   1


我用过:

def get_suborder_id(???)
     df['suborder_id'] = ????
     return df
df = df.groupby('orderid').apply(get_suborder_id)


但是,我不清楚它的实际工作方式。请帮助我理解这一点!!!

1 个答案:

答案 0 :(得分:1)

这更像cumcount

df['subid']=df.groupby('orderid').cumcount()+1