Проверьте, существует ли значение в индексе фрейма данных pandas

145

Я уверен, что есть очевидный способ сделать это, но сейчас не могу придумать ничего приятного.

В основном, вместо того, чтобы вызывать исключение, я хотел бы получить Trueили Falseувидеть, существует ли значение в dfиндексе pandas .

import pandas as pd
df = pd.DataFrame({'test':[1,2,3,4]}, index=['a','b','c','d'])
df.loc['g']  # (should give False)

Сейчас я работаю над следующим

sum(df.index == 'g')

python pandas ipython

— Абхи
источник

1

А что насчет любого (df.index == 'g')?

— luffe

264

Это должно сработать

'g' in df.index

— Гийом Жакено
источник

7

Это не работает, если несколько записей имеют одинаковые значения индекса.

— MaximG

2

@MaximG Что ты имеешь в виду? Это также работает для неуникального индекса.

— joris

Также работает с мультииндексом. Если ваш индекс имеет длину n, то 1..nможно проверить кортеж любой длины из

— Минь Триет

2

Для других, приходящих сюда, вам может потребоваться использовать, 'g' in df.columnsесли ваш фрейм данных был определен с заголовками столбцов, а не индексом, например:df = pandas.DataFrame({'test':[1,2,3,4]}, columns=['a','b','c','d'])

— Тахлор

3

Это постоянное время или линейное?

— Локеш

38

Просто для справки, поскольку это то, что я искал, вы можете проверить наличие в значениях или индексе, добавив метод ".values", например

g in df.<your selected field>.values
g in df.index.values

Я обнаружил, что добавление «.values» для получения простого списка или ndarray out делает существующие или «входящие» проверки более плавно с другими инструментами Python. Просто подумал, что брошу это людям.

— Иезекииль Круглик
источник

но AttributeError: объект DataFrame не имеет атрибута field

— Gank

1

Привет, Ганк. «Поле» должно было показать, что вы можете применить метод «.values» к различным полям фрейма данных, таким как столбцы или выбранный столбец. ".index" - это пример замены "field" фактическим доступным полем :) Думаю, это могло бы быть более понятным ...

— Иезекииль Круглик

2

Это было действительно полезно отметить. У меня есть иерархический случай, когда in g in df.indexвыдает истину и in g in df.index.valuesложь. Интересно.

— watsonic

@watsonic - одно предостережение, чтобы увидеть, возвращает ли один из них кортежи из-за иерархии. Обязательно посмотрите, что они выводят (например, в ipython или в командной строке), чтобы убедиться, что вы понимаете, с чем сравниваете. Еще одна вещь, которую вы можете сделать с иерархическими индексами, - это df.index.get_level_values (<level name>), чтобы сделать вещи более понятными - конечно, в зависимости от вашего приложения.

— Иезекииль Круглик

28

Мультииндекс работает немного иначе, чем одиночный индекс. Вот несколько методов для мультииндексированного фрейма данных.

df = pd.DataFrame({'col1': ['a', 'b','c', 'd'], 'col2': ['X','X','Y', 'Y'], 'col3': [1, 2, 3, 4]}, columns=['col1', 'col2', 'col3'])
df = df.set_index(['col1', 'col2'])

in df.index работает для первого уровня только при проверке одиночного значения индекса.

'a' in df.index     # True
'X' in df.index     # False

Проверьте df.index.levelsдругие уровни.

'a' in df.index.levels[0] # True
'X' in df.index.levels[1] # True

Проверьте df.indexкортеж с комбинацией индексов.

('a', 'X') in df.index  # True
('a', 'Y') in df.index  # False

— брокколи 2000
источник

2

с DataFrame: df_data

>>> df_data
  id   name  value
0  a  ampha      1
1  b   beta      2
2  c     ce      3

Я старался:

>>> getattr(df_data, 'value').isin([1]).any()
True
>>> getattr(df_data, 'value').isin(['1']).any()
True

но:

>>> 1 in getattr(df_data, 'value')
True
>>> '1' in getattr(df_data, 'value')
False

Так весело: D

— Sihc
источник

isinне будет проверять dtype. df['value'].isin([True]).any()попробуйте это, он также даст вам True, потому что он совпадает с 1. True -> 1.

— Mohamed Thasin ah

2

Код ниже не печатает логическое значение, но позволяет разбивать фреймы данных по индексу ... Я понимаю, что это, вероятно, не самый эффективный способ решения проблемы, но мне (1) нравится, как это читается, и (2) вы можете легко подмножество где индекс df1 существует в df2:

df3 = df1[df1.index.isin(df2.index)]

или где индекс df1 не существует в df2 ...

df3 = df1[~df1.index.isin(df2.index)]

— xxyjoel
источник

1

df = pandas.DataFrame({'g':[1]}, index=['isStop'])

#df.loc['g']

if 'g' in df.index:
    print("find g")

if 'isStop' in df.index:
    print("find a")

— Ганк
источник

Что такое isStop?

— Набин