具有 2 个或更多变量的 python pandas lambda

Question

我有一个数据框，我想在其中添加一个列，其中包含基于 2（可能是 3）个不同列中的条件的值的条件总和。我正在尝试使用 lambda 函数，例如：df['newColumn'] = df[['colA','colB']].apply(lambda x,y: df.loc[df['colA']==x].loc[df['colB']==y]['Total Amount'].sum())这种方法不起作用，尽管当我单独测试 .loc 语句并使用值代替 x 和 y 时，我确实得到了正确的总和。如果可能的话，我想引入另一列。我得到的错误是：“（）缺少 1 个必需的位置参数：'y'”，'发生在索引 colA。非常感谢任何帮助，

Answer 1

我的猜测是你想要这个：

df = pd.DataFrame({'A': [1,1,2,2,3,3],

                   'B': [2,2,2,3,3,3],

                   'TotalAmount': [10,20,30,40,50,60]})

df['NewColumn'] = df.groupby(['A', 'B'])['TotalAmount'].transform('sum')

df

#   A  B  TotalAmount  NewColumn

#0  1  2           10         30

#1  1  2           20         30

#2  2  2           30         30

#3  2  3           40         40

#4  3  3           50        110

#5  3  3           60        110