numpy.searchsorted

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

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

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

Параметры:

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

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

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

См. также

sort

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

histogram

Создаёт гистограмму из одномерных данных.

Примечания

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

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

Примеры

>>> 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])