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

Question

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

df = df_ori[[ 'FINAL', 'DUE', 'ID', 'NAME', 'BUSINESS 1', 'TAX 2', 'COUNT' ]] 当 df_ori 中的所有变量都存在时，这有效并且 df 存在。但是，如果缺少一个变量，我想通过管理潜在错误来更进一步。

if df.empty: print("Field(s) mendatory missing") #doesn't work if in df_ori dataframe the field 'ID' is missing 例如。我无法输入此“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'] not in index"采用，df.reindex(['A','B','C','Z'], axis=1)输出：   A  B  C   Z0  9  9  8 NaN1  2  6  7 NaN2  6  6  6 NaN3  3  7  9 NaN4  7  2  2 NaN