pandas.Index.searchsorted

Index.searchsorted(value, side='left', sorter=None) [source]

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

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

Параметры:

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

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

int или массив int Скаляр или массив точек вставки с той же формой, что и value. Изменено в версии 0.24.0: Если value является скаляром, всегда возвращается int. Ранее для скалярных входных данных возвращался массив из 1 элемента для Series и Categorical.

См. также

numpy.searchsorted

Примечания

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

Примеры

>>> x = pd.Series([1, 2, 3])
>>> x
0    1
1    2
2    3
dtype: int64

>>> x.searchsorted(4)
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'], ordered=True)
[apple, bread, bread, cheese, milk]
Categories (4, object): [apple < bread < cheese < milk]

>>> x.searchsorted('bread')
1

>>> x.searchsorted(['bread'], side='right')
array([3])