Generate Column Name Lookup in Python

Asker: ignoring_gravity

Photo by Alex Chumak on Unsplash

Disclaimer: Solution by ChatGPT and validated by human.

Question

Say I have

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   ┆…