Як перевірити, чи існує стовпець у Pandas

Чи є спосіб перевірити, чи існує стовпець у DataFrame Pandas?

Припустимо, у мене є така 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']                

Як в коментарі вказує @brianpck, set([])альтернативно можна побудувати фігурні дужки,

if {'A', 'C'}.issubset(df.columns):

Дивіться це запитання для обговорення синтаксису фігурних дужок.

Або ви можете використати розуміння списку, як у:

if all([item in df.columns for item in ['A','C']]):

Просто запропонувавши інший спосіб без використання операторів if, ви можете використовувати get()метод для DataFrames. Для виконання суми на основі запитання:

df['sum'] = df.get('A', df['B']) + df['C']

Метод DataFrameget має подібну поведінку, як і словники python.