链接聚合程序

Question

我有以下代码：

myData_agg = myData.groupby("Customer")["PurchAmount"].sum()
myData_agg.loc[myData_agg>=100,]

我可以在一个程序中编写此代码吗？ 谢谢！

Answer 1

选项一：

使用[]和lambda表达式链接它们：

myData.groupby("Customer")["PurchAmount"].sum()[lambda x: x >= 100]

选项二：

使用compress方法：

myData.groupby("Customer")["PurchAmount"].sum().compress(lambda x: x >= 100)

选项三：

使用pipe：

myData.groupby("Customer")["PurchAmount"].sum().pipe(lambda x: x[x >= 100])