Multi-Groupby(迭代或应用功能)

时间:2019-01-29 04:42:59

标签: python pandas function loops pandas-groupby

发布信息,因为这是我第一次无法找到解决问题的答案。我已经接近了,但是没有骰子。我正在尝试迭代,循环或将函数应用于按两列分组的pandas数据框,而我想要获取的数据在对第1列和第2列进行分组之后位于第三列。

给出示例数据:

df = pd.DataFrame({'Class':np.random.randint(1,10,100),'Type':np.random.choice(list('ABCD'),100),'Guid':np.random.randint(10000,99999,100)})

如果我使用

for name,group in df.groupby(['Class','Type']):
print(name)
print(group.Guid)

for name,group in df.groupby(['Class','Type'])['Guid']:
print(name)
print(group)

都可以达到我想要的输出

    (1, 'A')
86    86606
89    69999
Name: Guid, dtype: int32
(1, 'B')
0     71982
67    95343
68    84520
75    57036
87    61938
Name: Guid, dtype: int32
(1, 'C')
48    67008
Name: Guid, dtype: int32
(1, 'D')
4     87947
6     98477
13    78437
53    95792
76    66178
Name: Guid, dtype: int32
(2, 'A')
42    47960
Name: Guid, dtype: int32
(2, 'B')
30    69891
82    84436
91    54546
Name: Guid, dtype: int32
(3, 'A')
5     74182
19    12167
35    99939
78    38939
Name: Guid, dtype: int32
(3, 'B')
16    92239
27    25167
41    84245
47    43174
51    50472
Name: Guid, dtype: int32

我接下来要做的是一些数学运算,或者将除标准均值,标准差,计数等以外的函数应用到分组设置中的“导引”。这些工作正常,甚至可以在它们上面汇总。

df.groupby(['Class','Type']).Guid.mean()
Class  Type
1      A       78302.500000
       B       74163.800000
       C       67008.000000
       D       85366.200000
2      A       47960.000000
       B       69624.333333
3      A       56306.750000
       B       59059.400000
       C       31237.000000
       D       61973.000000
4      A       36573.000000
       B       55441.250000
       C       49381.333333
       D       35420.000000
5      A       57252.500000
       B       59267.250000
       C       48885.000000
       D       62163.000000
6      A       43106.800000
       B       46767.500000
       C       23170.000000
       D       31663.000000
7      A       47804.750000
       B       61570.666667
       C       60989.666667
       D       57300.000000
8      A       55890.000000
       B       50629.000000
       D       26312.000000
9      A       13338.500000
       B       49556.000000
       C       66602.400000
       D       58603.500000
Name: Guid, dtype: float64

但是我无法弄清楚当按['Class','Type']分组时,如何将自己的数学或函数应用于'Guid'。例如,如果我想计算“ Guid”中的值更改超过10,000的次数。

df.groupby(['Class','Type']).Guid.apply(function or maths here)
df.groupby(['Class','Type']).agg(['count','mean','std',somefunctionhere]).Guid

非常感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

这里有一个有关如何应用您自己的自定义功能的示例:

import pandas as pd
import numpy as np

def myfunc(group):
    # You can do whatever you want to the group, just make sure it returns that grouping as dataframe
    group['guid_sum'] = group['Guid'].sum()
    return group

df = pd.DataFrame({'Class':np.random.randint(1,10,100),'Type':np.random.choice(list('ABCD'),100),'Guid':np.random.randint(10000,99999,100)})
group = df.groupby(['Class','Type'])
applied = group.apply(myfunc)