Python:每组随机选择

发布于 2021-01-29 18:40:36

假设我有一个看起来像这样的数据框:

Name Group_Id
AAA  1
ABC  1
CCC  2
XYZ  2
DEF  3 
YYH  3

如何才能为每一行随机选择一个(或多个)行Group_Id?假设我希望每人随机抽取一次Group_Id,我将得到:

Name Group_Id
AAA  1
XYZ  2
DEF  3
关注者
0
被浏览
49
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。
    size = 2        # sample size
    replace = True  # with replacement
    fn = lambda obj: obj.loc[np.random.choice(obj.index, size, replace),:]
    df.groupby('Group_Id', as_index=False).apply(fn)
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看