Проверка наличия столбца в DataFrame библиотеки Pandas

Одним из распространенных вопросов при работе с библиотекой Pandas в Python является вопрос о том, как проверить, существует ли определенный столбец в DataFrame. Это может быть полезно, например, при выполнении операций над данными, когда необходимо исключить возможность ошибки из-за отсутствия нужного столбца.

Рассмотрим типичную ситуацию. Представим DataFrame, который состоит из трех столбцов: «A», «B» и «C».

   A   B    C
0  3  40  100
1  6  30  200

Необходимо создать новый столбец «sum», который будет равен сумме значений столбцов «A» и «C». Однако, если столбец «A» не существует, то в качестве альтернативы столбец «sum» должен быть равен сумме значений столбцов «B» и «C».

Как проверить наличие столбца?

Для проверки наличия столбца в DataFrame можно использовать оператор in. В примере ниже проверяется наличие столбца «A» в DataFrame df:

if 'A' in df:
    df['sum'] = df['A'] + df['C']
else:
    df['sum'] = df['B'] + df['C']

Таким образом, если столбец «A» существует в DataFrame, то он будет использован для расчета нового столбца «sum». В противном случае, будет использован столбец «B».

Этот подход позволяет избежать ошибок при выполнении операций над DataFrame, если отсутствует один из необходимых для расчетов столбцов.