python - groupby 并添加一些行

Question

下面有一个数据框

   A  B
0  a  1
1  a  2
2  c  3
3  c  4
4  e  5

我想得到下面的求和结果。key = column A

df.B.groupby(df.A).agg(np.sum)

但我想添加特定的行。

   B
a  3
b  0
c  7
d  0
e  5
f  0

但我应该添加行“b”和“d”。“f”

我怎样才能得到这个结果？

Answer 1

使用重新索引

df.groupby('A').B.sum().reindex(list('abcdef'), fill_value=0)

A
a    3
b    0
c    7
d    0
e    5
f    0
Name: B, dtype: int64