numpy.searchsorted

numpy.searchsorted(a, v, side='left', sorter=None) [source]

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

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

Предполагая, что a отсортирован:

side	возвращённый индекс i удовлетворяет
left	a[i-1] < v <= a[i]
right	a[i-1] <= v < a[i]

Параметры:

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

Возвращает:

indices : array of ints Массив точек вставки с той же формой, что и v.

См. также

sort

Возвращает отсортированную копию массива.

histogram

Вычисляет гистограмму из одномерных данных.

Примечания

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

Начиная с NumPy 1.4.0 searchsorted работает с вещественными/комплексными массивами, содержащими значения nan. Улучшенный порядок сортировки описан в sort.

Эта функция является более быстрой версией встроенных функций python bisect.bisect_left (side='left') и bisect.bisect_right (side='right'), которая также векторизована в v аргументе.

Примеры

>>> np.searchsorted([1,2,3,4,5], 3)
2
>>> np.searchsorted([1,2,3,4,5], 3, side='right')
3
>>> np.searchsorted([1,2,3,4,5], [-10, 10, 2, 3])
array([0, 5, 1, 2])