python - Pandas 交叉表中的加权结果

Question

我想使用第三列来加权 pandas 交叉表中的结果。

例如，以下内容:

import pandas as pd
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'bar'],
                   'B': [1, 1, 0, 0, 0],
                   'weight': [2, 3, 4, 5, 6]})
print(pd.crosstab(df.A, df.B))

结果:

B    0  1
A        
bar  2  1
foo  1  1

我想要的结果是:

B     0  1
A        
bar  11  3
foo   4  2

Answer 1

您可以使用 aggfunc 参数为交叉表提供自定义聚合函数:

pd.crosstab(df.A, df.B, df.weight, aggfunc = sum)
B     0  1
A         
bar  11  3
foo   4  2