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定义
答案 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()