pandas.CategoricalIndex.searchsorted

CategoricalIndex.searchsorted(key, side='left', sorter=None)

Нахождение индексов, куда следует вставить элементы для сохранения порядка.

Нахождение индексов в отсортированном IndexOpsMixin self таким образом, что, если соответствующие элементы в v были вставлены перед индексами, порядок self сохранялся бы.

Параметры:

key : array_like Значения для вставки в self. side : {‘left’, ‘right’}, optional Если ‘left’, возвращается индекс первого подходящего местоположения. Если ‘right’, возвращается последний такой индекс. Если подходящего индекса нет, возвращается либо 0, либо N (где N — длина self). sorter : 1-D array_like, optional Необязательный массив целочисленных индексов, сортирующих self в порядке возрастания. Обычно они являются результатом np.argsort.

Возвращает:

indices : массив целых чисел Массив точек вставки с теми же размерами, что и v.

См. также

numpy.searchsorted

Примечания

Для поиска необходимых точек вставки используется двоичный поиск.

Примеры

>>> x = pd.Series([1, 2, 3])
>>> x
0    1
1    2
2    3
dtype: int64
>>> x.searchsorted(4)
array([3])
>>> x.searchsorted([0, 4])
array([0, 3])
>>> x.searchsorted([1, 3], side='left')
array([0, 2])
>>> x.searchsorted([1, 3], side='right')
array([1, 3])
>>>
>>> x = pd.Categorical(['apple', 'bread', 'bread', 'cheese', 'milk' ])
[apple, bread, bread, cheese, milk]
Categories (4, object): [apple < bread < cheese < milk]
>>> x.searchsorted('bread')
array([1])     # Note: an array, not a scalar
>>> x.searchsorted(['bread'])
array([1])
>>> x.searchsorted(['bread', 'eggs'])
array([1, 4])
>>> x.searchsorted(['bread', 'eggs'], side='right')
array([3, 4])    # eggs before milk