python - 如果数据框中不存在，如何在 Pandas 中添加列

Question

我在 python 中有以下列名数组。

 col = ['code','ro','quantity','amount']

我从一个可能有也可能没有上述所有列的函数中得到处理过的数据框。如果数组中缺少它们，我想添加默认值为 0 的列。

例如我得到以下数据框

df1

code      ro      quantity
123       342     34.56
123       445     54.56

我想要的数据框是

code      ro      quantity  amount
123       342     34.56     0
123       445     54.56     0

我怎样才能在 pandas 中做到这一点？

Answer 1

使用reindex在 axis=1 上:

df.reindex(col,fill_value=0,axis=1)

---

   code   ro  quantity  amount
0   123  342     34.56       0
1   123  445     54.56       0