1. ホーム
  2. パイソン

[解決済み】Pandasでカラムが存在するかどうかを確認する方法

2022-03-23 18:49:27

質問

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: