Есть ли способ проверить, существует ли столбец в 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 not 'A' in df.columns:
для выполнения операции, еслиA
ее нет вdf
Чтобы проверить, существует ли один или несколько столбцов, вы можете использовать
set.issubset
, как в:Как указывает @brianpck в комментарии, в качестве
set([])
альтернативы можно построить с помощью фигурных скобок,Смотрите этот вопрос для обсуждения синтаксиса фигурных скобок.
Или вы можете использовать понимание списка, как в:
источник
{'A', 'B'}.issubset()
Просто чтобы предложить другой путь без использования операторов if, вы можете использовать
get()
метод дляDataFrame
s. Для выполнения суммы по вопросу:Метод
DataFrame
get работает аналогично словарям python.источник