pandas分组依据并分配组id，然后取消分组

Question

我有一个以下格式的大数据集：

id, socialmedia
1, facebook
2, facebook
3, google
4, google
5, google
6, twitter
7, google
8, twitter
9, snapchat
10, twitter
11, facebook

我想分组，然后分配一个group_id列，然后将其解组（展开）回单个记录。在

^{pr2}$

我尝试了以下操作，但结果是“DataFrameGroupBy”对象不支持项分配。在

x['grpId'] = x.groupby('socialmedia')['socialmedia'].rank(method='dense').astype(int)

Answer 1

您可以使用sklearn.preprocessing.LabelEncoder方法：

In [79]: from sklearn.preprocessing import LabelEncoder

In [80]: le = LabelEncoder()

In [81]: df['groupId'] = le.fit_transform(df['socialmedia'])+1

In [82]: df
Out[82]:
    id socialmedia  groupId
0    1    facebook        1
1    2    facebook        1
2    3      google        2
3    4      google        2
4    5      google        2
5    6     twitter        4
6    7      google        2
7    8     twitter        4
8    9    snapchat        3
9   10     twitter        4
10  11    facebook        1