Сортировать кадр данных поляров по индексам в столбце ⇐ Python

[Anonymous]

Скажем, у меня есть:

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

df = pl.from_repr("""
shape: (4, 4)
┌─────┬─────┬─────┬──────┐
│ idx ┆ a   ┆ b   ┆ idx2 │
│ --- ┆ --- ┆ --- ┆ ---  │
│ u32 ┆ i64 ┆ i64 ┆ u32  │
╞═════╪═════╪═════╪══════╡
│ 0   ┆ 1   ┆ 4   ┆ 3    │
│ 2   ┆ 1   ┆ 3   ┆ 0    │
│ 3   ┆ 2   ┆ 1   ┆ 2    │
│ 4   ┆ 2   ┆ 2   ┆ 4    │
└─────┴─────┴─────┴──────┘
""")

Я хотел бы отсортировать его так, чтобы столбец «idx» соответствовал порядку, заданному «idx2».
Желаемый результат:
p>

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

shape: (4, 4)
┌─────┬─────┬─────┬──────┐
│ idx ┆ a   ┆ b   ┆ idx2 │
│ --- ┆ --- ┆ --- ┆ ---  │
│ u32 ┆ i64 ┆ i64 ┆ u32  │
╞═════╪═════╪═════╪══════╡
│ 3   ┆ 2   ┆ 1   ┆ 2    │
│ 0   ┆ 1   ┆ 4   ┆ 3    │
│ 2   ┆ 1   ┆ 3   ┆ 0    │
│ 4   ┆ 2   ┆ 2   ┆ 4    │
└─────┴─────┴─────┴──────┘

Как это сделать, чтобы оно работало как в ленивом, так и в активном режиме?


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