polars.Series.search_sorted

Series.search_sorted(
    element: IntoExpr | np.ndarray[Any,
    Any] | None,
    side: SearchSortedSide = 'any',
    *,
    descending: bool = False,
) → int | Series

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

\[a[i-1] < v <= a[i]\]

Параметры:

element

Выражение или скалярное значение.

side{‘any’, ‘left’, ‘right’}

Если ‘any’, возвращается индекс первой найденной подходящей позиции. Если ‘left’, возвращается индекс самой левой найденной подходящей позиции. Если ‘right’, возвращается индекс самой правой найденной подходящей позиции.

descending

Логическое значение, указывающее, упорядочены ли значения по убыванию (в любом случае они должны быть отсортированы).

Примеры

>>> s = pl.Series("set", [1, 2, 3, 4, 4, 5, 6, 7])
>>> s.search_sorted(4)
3
>>> s.search_sorted(4, "left")
3
>>> s.search_sorted(4, "right")
5
>>> s.search_sorted([1, 4, 5])
shape: (3,)
Series: 'set' [u32]
[
    0
    3
    5
]
>>> s.search_sorted([1, 4, 5], "left")
shape: (3,)
Series: 'set' [u32]
[
    0
    3
    5
]
>>> s.search_sorted([1, 4, 5], "right")
shape: (3,)
Series: 'set' [u32]
[
    1
    5
    6
]