python中dataframe 判断是否存在_python – 如何检查列中是否存在Pandas

有一种方法来检查一个Pandas DataFrame中是否存在列？

假设我有以下DataFrame：

>>> import pandas as pd

>>> from random import randint

>>> df = pd.DataFrame({'A': [randint(1, 9) for x in xrange(10)],

'B': [randint(1, 9)*10 for x in xrange(10)],

'C': [randint(1, 9)*100 for x in xrange(10)]})

>>> df

A B C

0 3 40 100

1 6 30 200

2 7 70 800

3 3 50 200

4 7 50 400

5 4 10 400

6 3 70 500

7 8 30 200

8 3 40 800

9 6 60 200

我想计算df [‘sum’] = df [‘A’] df [‘C’]

但是首先我想检查是否存在df [‘A’]，如果不存在，我想要计算df [‘sum’] = df [‘B’] df [‘C’]。

谢谢您的帮助。