polars.Expr.search_sorted

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

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

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

движок:В памяти

Параметры:

element

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

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

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

descending

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

Примеры

>>> df = pl.DataFrame(
...     {
...         "values": [1, 2, 3, 5],
...     }
... )
>>> df.select(
...     [
...         pl.col("values").search_sorted(0).alias("zero"),
...         pl.col("values").search_sorted(3).alias("three"),
...         pl.col("values").search_sorted(6).alias("six"),
...     ]
... )
shape: (1, 3)
┌──────┬───────┬─────┐
│ zero ┆ three ┆ six │
│ ---  ┆ ---   ┆ --- │
│ u32  ┆ u32   ┆ u32 │
╞══════╪═══════╪═════╡
│ 0    ┆ 2     ┆ 4   │
└──────┴───────┴─────┘