Скажем, у меня есть это:
>>> df = polars.DataFrame(dict(j=[1,1,1,2,1,2,1,2,2], k=[1,2,3,4,5,6,7,8,9]))
>>> df
shape: (9, 2)
┌─────┬─────┐
│ j ┆ k │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1 ┆ 1 │
│ 1 ┆ 2 │
│ 1 ┆ 3 │
│ 2 ┆ 4 │
│ 1 ┆ 5 │
│ 2 ┆ 6 │
│ 1 ┆ 7 │
│ 2 ┆ 8 │
│ 2 ┆ 9 │
└─────┴─────┘
Есть ли более короткий/лучший способ имитировать dropWhile (т. е. удалить все строки в верхней части DataFrame, которые удовлетворяют определенному условию), чем этот (который удаляет все начальные строки с j == 1):
>>> df.slice(df.with_row_index().filter(polars.col('j') != 1).select('index').head(1).item())
shape: (6, 2)
┌─────┬─────┐
│ j ┆ k │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 2 ┆ 4 │
│ 1 ┆ 5 │
│ 2 ┆ 6 │
│ 1 ┆ 7 │
│ 2 ┆ 8 │
│ 2 ┆ 9 │
└─────┴─────┘
Подробнее здесь: https://stackoverflow.com/questions/762 ... -in-polars
