选择每个类的n元素

时间:2017-01-25 18:21:40

标签: python pandas sklearn-pandas

我使用大熊猫,我有一组数据,大约有4百万的观察结果。 我想知道为每个类选择50个随机元素或前50个元素的最佳/最快/最有效的方法是什么(类只是一列)。

我列中的唯一类数约为2k,我想选择100,000个元素的子集,每个类50个元素。

我正考虑将它们分组到类中,然后遍历每个组并选择前50个元素,然后继续下一组。

我想知道有更好的方法吗?

2 个答案:

答案 0 :(得分:1)

IIUC您需要使用numpy.random.choice

import pandas as pd
import numpy as np 

df = pd.DataFrame({'class': [0,1,2,3,0,1,2,3,0,1,2,3],
               'value': [1,2,3,4,5,6,7,8,9,10,1,12]})


Samplesize = 2  #number of samples that you want       
print df.groupby('class', as_index=False).apply(lambda array: array.loc[np.random.choice(array.index, Samplesize, False),:])

输入

    class  value
0       0      1
1       1      2
2       2      3
3       3      4
4       0      5
5       1      6
6       2      7
7       3      8
8       0      9
9       1     10
10      2      1
11      3     12

输出

      class  value
0 8       0      9
  0       0      1
1 1       1      2
  5       1      6
2 6       2      7
  10      2      1
3 11      3     12
  3       3      4

答案 1 :(得分:1)

给出以下数据框

df = pd.DataFrame(np.random.rand(100, 2), columns=list('ab'))
df['group'] = np.remainder(np.random.permutation(len(df)), 3)

df.head()

    a           b           group
0   0.069140    0.553955    1
1   0.564991    0.699645    2
2   0.251304    0.516667    2
3   0.962819    0.314219    2
4   0.353382    0.500961    0

您可以通过

获得随机版本
df_randomized = df.ix[np.random.permutation(len(df))]

df_randomized.head()

    a           b           group
90  0.734971    0.895469    0
35  0.195013    0.566211    0
27  0.370124    0.870052    2
21  0.297194    0.500713    1
66  0.319668    0.347365    2

要选择N个随机元素,首先生成排列并减小其大小。之后将其应用于数据帧:

N = 10
indexes = np.random.permutation(len(df))[:N]
df_randomized = df.ix[indexes]

要获取每个组的前N个元素,您可以对数据帧进行分组并应用方法来选择前N个元素。这里不需要任何循环,因为pandas可以为你处理:

N = 10
df.groupby('group')\
    .apply(lambda x: x[:N][['a', 'b']])

所有这些方法应该很快,因为它们使用numpy或pandas的内部优化方法。