Скажем, у меня есть
df = pl.DataFrame({
'a': [1, 2, 1],
'b': [2, 1, 2],
'c': [3, 3, 2],
'column': ['a', 'c', 'b'],
})
shape: (3, 4)
┌─────┬─────┬─────┬────────┐
│ a ┆ b ┆ c ┆ column │
│ --- ┆ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 ┆ str │
╞═════╪═════╪═════╪════════╡
│ 1 ┆ 2 ┆ 3 ┆ a │
│ 2 ┆ 1 ┆ 3 ┆ c │
│ 1 ┆ 2 ┆ 2 ┆ b │
└─────┴─────┴─────┴────────┘
Я хочу добавить столбец, который для каждой строки будет принимать значение из строки, соответствующей столбцу в столбце.
Ожидаемый результат:
shape: (3, 5)
┌─────┬─────┬─────┬────────┬────────┐
│ a ┆ b ┆ c ┆ column ┆ lookup │
│ --- ┆ --- ┆ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 ┆ str ┆ i64 │
╞═════╪═════╪═════╪════════╪════════╡
│ 1 ┆ 2 ┆ 3 ┆ a ┆ 1 │
│ 2 ┆ 1 ┆ 3 ┆ c ┆ 3 │
│ 1 ┆ 2 ┆ 2 ┆ b ┆ 2 │
└─────┴─────┴─────┴────────┴────────┘
Подробнее здесь: https://stackoverflow.com/questions/766 ... -in-column
