python - Pandas:创建具有零值的缺失组合行

Question

假设我有一个数据框 df:

df = pd.DataFrame({'col1': [1,1,2,2,2], 'col2': ['A','B','A','B','C'], 'value': [2,4,6,8,10]})

    col1 col2 value
0   1    A    2
1   1    B    4
2   2    A    6
3   2    B    8
4   2    C    10

我正在寻找一种方法，在 col1 和 col2 与现有值的可能组合中创建任何缺失的行，并用零填充缺失的行

期望的结果是:

    col1 col2 value
0   1    A    2
1   1    B    4
2   2    A    6
3   2    B    8
4   2    C    10
5   1    C    0    <- Missing the "1-C" combination, so create it w/ value = 0

我已经研究过使用 stack 和 unstack 来完成这项工作，但我不确定这是否正是我所需要的。

提前致谢

Answer 1

使用pivot，然后使用stack

df.pivot(*df.columns).fillna(0).stack().to_frame('values').reset_index()
Out[564]: 
   col1 col2  values
0     1    A     2.0
1     1    B     4.0
2     1    C     0.0
3     2    A     6.0
4     2    B     8.0
5     2    C    10.0