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

[Anonymous]

У меня есть фрейм данных

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

import polars as pl
import numpy as np

df = pl.DataFrame(
{
"nrs": [1, 2, 3, None, 5],
"names": ["foo", "ham", "spam", "egg", None],
"random": np.random.rand(5),
"groups": ["A", "A", "B", "C", "B"],
}
)

Я хочу выбрать только столбцы в моем списке:

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

mylist = ['nrs', 'random']

Кажется, это работает:

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

import polars.selectors as cs

df.select(cs.by_name(mylist)))

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

shape: (5, 2)
┌──────┬──────────┐
│ nrs  ┆ random   │
│ ---  ┆ ---      │
│ i64  ┆ f64      │
╞══════╪══════════╡
│ 1    ┆ 0.727509 │
│ 2    ┆ 0.14311  │
│ 3    ┆ 0.244337 │
│ null ┆ 0.603872 │
│ 5    ┆ 0.222068 │
└──────┴──────────┘

Это идиоматический способ сделать это? Или есть лучшие способы?


Подробнее здесь: https://stackoverflow.com/questions/779 ... -dataframe