python - 在数据框中选择变量时管理结果缺失

Question

当我的数据集包含所有变量时，我可以通过仅选择我需要的变量来创建子集。但如果缺少一个，我就一无所获。我该如何处理这样的错误？

df = df_ori[[ '最终的'， '到期的'， 'ID'， '姓名'， '业务 1', “税2”， '数数' ]]当 df_ori 中的所有变量都存在时，这有效并且 df 存在。但如果缺少一个变量，我想进一步管理潜在的错误。

if df.empty: print("Field(s) mendatory Missing") #如果在 df_ori 数据帧中缺少字段“ID”，则不起作用。我无法输入这个“IF”

没有错误，但没有生成数据帧 df 或未出现“字段强制缺失”

Answer 1

IIUC，使用 reindex 和 axis=1:

df = pd.DataFrame(np.random.randint(0,10,(5,5)), columns=[*'ABCDE'])

哪里，

df[['A','B','C','Z']]

生成KeyError:“['Z']不在索引中”

使用，

df.reindex(['A','B','C','Z'], axis=1)

输出:

   A  B  C   Z
0  9  9  8 NaN
1  2  6  7 NaN
2  6  6  6 NaN
3  3  7  9 NaN
4  7  2  2 NaN