python随机分组的方法_Python Pandas从Groupby中选择组的随机样本

您可以获取df.some_key.unique()的唯一值的randoms示例,使用它来对df进行切片,最后对结果进行groupby：

In [337]:

df = pd.DataFrame({'some_key': [0,1,2,3,0,1,2,3,0,1,2,3],

'val': [1,2,3,4,1,5,1,5,1,6,7,8]})

In [338]:

print df[df.some_key.isin(random.sample(df.some_key.unique(),2))].groupby('some_key').mean()

val

some_key

0 1.000000

2 3.666667

如果有多个groupby键：

In [358]:

df = pd.DataFrame({'some_key1':[0,1,2,3,0,1,2,3,0,1,2,3],

'some_key2':[0,0,0,0,1,1,1,1,2,2,2,2],

'val': [1,2,3,4,1,5,1,5,1,6,7,8]})

In [359]:

gby = df.groupby(['some_key1', 'some_key2'])

In [360]:

print gby.mean().ix[random.sample(gby.indices.keys(),2)]

val

some_key1 some_key2

1 1 5

3 2 8

但是如果你只想获得每个组的值,你甚至不需要麻烦,MultiIndex会做：

In [372]:

idx = random.sample(set(pd.MultiIndex.from_product((df.some_key1, df.some_key2)).tolist()),

2)

print df.set_index(['some_key1', 'some_key2']).ix[idx]

val

some_key1 some_key2

2 0 3

3 1 5