问题描述
有没有办法检查一个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']
。
解决方法
第一种:
这将工作:
if 'A' in df:
但为了清楚起见,我可能会将它写为:
if 'A' in df.columns:
第二种:
要检查是否存在一列或多列,可以使用set.issubset
,如下所示:
if set(['A','C']).issubset(df.columns):
df['sum'] = df['A'] + df['C']
set([])
也可以用大括号来构造:
if {'A', 'C'}.issubset(df.columns):