熊猫：groupby之后的样本组

Question

在大熊猫分组之后如何采样分组？ 假设我要在groupby之后获得分组的前半部分。

In [194]: df = pd.DataFrame({'name':['john', 'george', 'john','andrew','Daniel','george','andrew','Daniel'], 'hits':[12,34,13,23,53,47,20,48]})
In [196]: grouped = df.groupby('name')

有“ john”，“ george”，“ andrew”，“ daniel” 4个小组分组 ，我有兴趣从4个小组中分出2个小组。无论返回哪2个小组都没关系。

非常感谢你。

Answer 1

您可以提前对名称进行采样，并且仅将所选名称分组：

selected_names = np.random.choice(df.name.unique(),2,replace = False)
grouped = df[df.name.isin(selected_names)].groupby('name')

Answer 2

感谢您的快速答复，ajcr和cwharland。 我可能不清楚我想要什么，但您的建议很好。 我做了：

choices =np.random.choice(grouped.indices.keys(), 2, replace=False)
df[df['name'].isin(choices)]

并得到我希望的结果：

Out[215]: 
   hits    name
0    12    john
2    13    john
3    23  andrew
6    20  andrew

谢谢你俩！