Python 3.x 如何在透视表中使用多列值计算结果

我正在按年份对数据帧进行分组，并希望找到B列的加权平均值
我假设将A列和B列相加，然后将结果除以A的和[functian WA（）这样做]

我真的不知道如何通过传递这两个值来调用函数

def WA(A,B):
    c = A*B
    return c.sum()/A.sum()
    
sample = pd.pivot_table(intime,index=['year'], columns = 'DV', values = ['A','B'],
                        aggfunc={'A' : [len,np.sum], 'B' : [WA]}, fill_value = 0)