如何检查 Pandas 中是否存在列

新手上路,请多包涵

如何检查 Pandas DataFrame df 中是否存在列?

    A   B    C
0  3  40  100
1  6  30  200

我将如何检查 "A" 列是否存在于上述 DataFrame 中,以便我可以计算:

 df['sum'] = df['A'] + df['C']

如果 "A" 不存在:

 df['sum'] = df['B'] + df['C']

原文由 npires 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 695
1 个回答

这将起作用:

 if 'A' in df:

但为了清楚起见,我可能会写成:

 if 'A' in df.columns:

原文由 chrisb 发布,翻译遵循 CC BY-SA 3.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题