Question

我正在寻找一种链接groupby并应用的方法，如下所示（参见下面的代码以获取具体示例）：

df.groupby("a").apply(func_1).groupby("b").apply(func_2)

我想它不起作用，因为groupby需要输入一个数据帧，这并不总是上面第二组的情况（可以输入一个系列，参见例子）。一个解决方案可能是让第一个应用输出func_1的结果加上原始数据帧，但我还没有找到如何做到这一点。

我正在寻找一般的解决方法，而不仅仅是针对此特定示例的解决方法。

示例：假设我想计算b中每个组的a的路缘下面积，然后计算c中每个组的这些面积的总和。

df=pd.DataFrame({"a":np.arange(8),"b":np.repeat(np.arange(4),2),
"c":np.repeat(np.arange(2),4)})

df
   a  b  c
0  0  0  0
1  1  0  0
2  2  1  0
3  3  1  0
4  4  2  1
5  5  2  1
6  6  3  1
7  7  3  1


df.groupby("b").apply(lambda x: trapz(x["a"])).groupby("c").apply(sum)   
Traceback (most recent call last):
[...]
KeyError: 'c'


#Expected output
c
0     3.0
1    11.0


#I know that this code works, but I would like to avoid to modify 
#my dataframe :

df["result"]=list(df
    .groupby("b").apply(lambda x: trapz(x["a"]))
    .repeat(df.groupby("b").size()))
df.groupby("b").first().groupby("c").result.sum()

任何帮助非常感谢！

Answer 1

我想我会做类似的事情：

# your_fun is the function you want to apply
df.groupby('c').apply(lambda f: sum(f.groupby('b')['a'].apply(your_fun))

链接groupby并应用pandas

1 个答案: