Полярные значения — получить значение столбца с минимальным/максимальным значением другого столбца. ⇐ Python

[Anonymous]

Учитывается следующий полярный кадр данных:

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

pl.DataFrame({'A': ['a0', 'a0', 'a1', 'a1'],
'B': ['b1', 'b2', 'b1', 'b2'],
'x': [0, 10, 5, 1]})

shape: (4, 3)
┌─────┬─────┬─────┐
│ A   ┆ B   ┆ x   │
│ --- ┆ --- ┆ --- │
│ str ┆ str ┆ i64 │
╞═════╪═════╪═════╡
│ a0  ┆ b1  ┆ 0   │
│ a0  ┆ b2  ┆ 10  │
│ a1  ┆ b1  ┆ 5   │
│ a1  ┆ b2  ┆ 1   │
└─────┴─────┴─────┘

Я хочу добавить столбец y , какие группы по и выбирают значение из b с максимальным соответствующим x . Следующим раме данных должен быть результатом: < /p>

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

┌─────┬─────┬─────┬─────┐
│ A   ┆ B   ┆ x   ┆ y   │
│ --- ┆ --- ┆ --- ┆ --- │
│ str ┆ str ┆ i64 ┆ str │
╞═════╪═════╪═════╪═════╡
│ a0  ┆ b1  ┆ 0   ┆ b2  │
│ a0  ┆ b2  ┆ 10  ┆ b2  │
│ a1  ┆ b1  ┆ 5   ┆ b1  │
│ a1  ┆ b2  ┆ 1   ┆ b1  │
└─────┴─────┴─────┴─────┘

Я пробовал различные версии df.with_columns (y = pl.col ('b').?. Over ('a')) без какой -либо удачи.

Подробнее здесь: https://stackoverflow.com/questions/793 ... -max-value