python - 如果在 pandas df 中丢失，则创建行

Question

当前数据:

  |ID | DT     | STATE | V|
  |1  | 201901 | PA    | 1|
  |1  | 201902 | PA    | 6|
  |2  | 201902 | PA    | 3|
  |1  | 201902 | CA    | 3|
  |2  | 201901 | CA    | 1|

我想用 ID 的所有组合创建行, DT和 STATE与 V在它不可用的情况下为 0 :

  |ID | DT     | STATE | V|
  |1  | 201901 | PA    | 1|
  |1  | 201902 | PA    | 6|
  |1  | 201901 | CA    | 0|
  |1  | 201902 | CA    | 3|
  |2  | 201901 | PA    | 0|
  |2  | 201902 | PA    | 3|
  |2  | 201901 | CA    | 1|
  |2  | 201902 | CA    | 0|

谢谢

Answer 1

你可以做MultiIndex然后索引 reindex

idx=pd.MultiIndex.from_product([df.ID.unique(),df.DT.unique(),df.STATE.unique()])
df=df.set_index(['ID','DT','STATE']).reindex(idx,fill_value=0).reset_index()
df
   level_0  level_1 level_2  V
0        1   201901      PA  1
1        1   201901      CA  0
2        1   201902      PA  6
3        1   201902      CA  3
4        2   201901      PA  0
5        2   201901      CA  1
6        2   201902      PA  3
7        2   201902      CA  0