Pandas aggragation error(需要一个整数)

时间:2017-12-11 15:11:10

标签: python python-3.x pandas-groupby

df_act = pd.DataFrame({'A': {0: 'CHEMBL264', 1: 'CHEMBL4124', 2: 'CHEMBL264', 3: 'CHEMBL233', 4: 'CHEMBL233', 5: 'CHEMBL237', 6: 'CHEMBL236', 7: 'CHEMBL312', 8: 'CHEMBL3820', 9: 'CHEMBL3820'}, 'B': {0: 8.6999999999999993, 1: 8.1600000000000001, 2: 8.3000000000000007, 3: 7.2400000000000002, 4: 8.0, 5: 6.1600000000000001, 6: 6.4400000000000004, 7: 4.8200000000000003, 8: 7.5899999999999999, 9: 7.4299999999999997}})

这样做有效:

df_act.groupby(['A'])['B'].median()

但是,使用自定义函数将其应用于groupby对象会失败:

def fun(x):
     name = {'B_median': x['B'].median()}
     return(pd.Series(names, index = ['B_median']))

df_act.groupby(['A'])['B'].apply(fun)

返回:

    ---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc (pandas/_libs/index.c:5126)()

pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.Int64HashTable.get_item (pandas/_libs/hashtable.c:14010)()

TypeError: an integer is required

当然,在这两个示例中,我使用相同的数据帧,因此我不理解错误。

编辑:添加df_act定义

1 个答案:

答案 0 :(得分:2)

问题在于,在此示例中,您需要更改

df_act.groupby(['A'])['B'].apply(fun)

df_act.groupby(['A']).apply(fun)

详见How is pandas groupby method actually working?.apply的要点就是将函数应用于每个"子数据框" (组),然后将每个组的结果重新组合到您的结果中。

fun中,您已经引用了' B'。因此事先对其进行索引是多余的。

另请注意,您并不需要将返回的对象包装在系列中。它仍然有点做作,但这就足够了:

def fun(x):
     return x['B'].median()