python - 在组内添加缺失的行

Question

假设数据框:

df = pd.DataFrame({
    "id":     [1, 1, 1, 2, 2, 3],
    "day":    [1, 2, 3, 1, 3, 2],
    "value":  [1, 2, 3, 4, 5, 6],
})

我需要所有 ID 具有相同的日期集。如何添加缺失日期的行？

Answer 1

国际联合会

df=df.pivot(*df.columns).fillna(0).stack().reset_index()
   id  day    0
0   1    1  1.0
1   1    2  2.0
2   1    3  3.0
3   2    1  4.0
4   2    2  0.0
5   2    3  5.0
6   3    1  0.0
7   3    2  6.0
8   3    3  0.0