Pandas groupby:百分比高于阈值

时间:2015-02-23 21:09:41

标签: python pandas

我有一个DataFrame,我希望使用groupby,但我正在寻找一些不寻常的功能来聚合。我想让每组中的观察百分比高于某个阈值。例如,阈值为0,DataFrame

df = pd.DataFrame(dict(day=[1, 1, 1, 2, 2, 2, 3, 3, 3, 4], value=[0, 4, 0, 4, 0, 4, 0, 4, 0, 4]))

df
   day  value
0    1      0
1    1      4
2    1      0
3    2      4
4    2      0
5    2      4
6    3      0
7    3      4
8    3      0
9    4      4

应该成为

df_group = pd.DataFrame(dict(day=[1, 2, 3, 4], value=[.33, .67, .33, 1.0]))

df_group
   day  value
0    1   0.33
1    2   0.67
2    3   0.33
3    4   1.00

我也在使用相当大的数据集,所以我很感激将计算时间考虑在内。

1 个答案:

答案 0 :(得分:11)

>>> df.groupby('day')['value'].apply(lambda c: (c>0).sum()/len(c))
day
1      0.333333
2      0.666667
3      0.333333
4      1.000000
Name: value, dtype: float64