Подсчет перекрывающихся вхождений подстроки в строках в Polars ⇐ Python

[Anonymous]

Как лучше всего подсчитать количество перекрывающихся вхождений заданной подстроки в строках, хранящихся в столбце Polars DataFrame?
API Polars предоставляет функцию Polars.Expr.str.count_matches, которая подсчитывает все последовательные непересекающиеся совпадения регулярных выражений.
Например:

Код: Выделить всё

import polars as pl

df = pl.DataFrame({"foo": ["aaaaa", "aabaa", "aaaab"]})

df.with_columns(pl.col("foo").str.count_matches("aa"))

Код: Выделить всё

shape: (3, 1)
┌─────┐
│ foo │
│ --- │
│ u32 │
╞═════╡
│ 2   │
│ 2   │
│ 2   │
└─────┘

Как получить количество перекрывающихся вхождений? Ожидаемый результат в этом случае — [4,2,3].


Подробнее здесь: https://stackoverflow.com/questions/780 ... -in-polars