pandas.Index.isin

Index.isin(values, level=None)[source]

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

Вычисляет булевый массив, определяющий, содержится ли каждое значение индекса в переданном наборе значений. Длина возвращаемого булевого массива соответствует длине индекса.

Параметры:

values:множество или список-подобный объект

Искомые значения.

level:строка или целое число, необязательно

Имя или позиция уровня индекса для использования (если индекс является MultiIndex).

Возвращаемое значение:

np.ndarray[bool]

Массив NumPy с булевыми значениями.

См. также

Series.isin

То же самое для Series.

DataFrame.isin

Тот же метод для DataFrames.

Примечания

В случае MultiIndex вы должны либо указать values как список-подобный объект, содержащий кортежи, длина которых соответствует количеству уровней, или указать level. В противном случае будет вызвано ValueError.

Если указан level:

• если это имя одного и только одного уровня индекса, используйте этот уровень;

• в противном случае это должно быть число, указывающее позицию уровня.

Примеры

>>> idx = pd.Index([1,2,3])
>>> idx
Index([1, 2, 3], dtype='int64')

Проверьте, содержится ли каждое значение индекса в списке значений.

>>> idx.isin([1, 4])
array([ True, False, False])

>>> midx = pd.MultiIndex.from_arrays([[1,2,3],
...                                  ['red', 'blue', 'green']],
...                                  names=('number', 'color'))
>>> midx
MultiIndex([(1,   'red'),
            (2,  'blue'),
            (3, 'green')],
           names=['number', 'color'])

Проверьте, содержатся ли строки в уровне «цвет» MultiIndex в списке цветов.

>>> midx.isin(['red', 'orange', 'yellow'], level='color')
array([ True, False, False])

Для проверки по всем уровням MultiIndex передайте список кортежей:

>>> midx.isin([(1, 'red'), (3, 'red')])
array([ True, False, False])