我有一个数据框
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"],
}
)
我想只 Select list中的列:
mylist = ['nrs', 'random']
这似乎奏效了:
import polars.selectors as cs
df.select(cs.by_name(mylist)))
这是惯用的做法吗？还是有更好的办法？
实际上要简单得多:
df.select(['nrs', 'random'])
┌──────┬──────────┐
│ nrs ┆ random │
│ --- ┆ --- │
│ i64 ┆ f64 │
╞══════╪══════════╡
│ 1 ┆ 0.662732 │
│ 2 ┆ 0.437345 │
│ 3 ┆ 0.43857 │
│ null ┆ 0.701177 │
│ 5 ┆ 0.390494 │
└──────┴──────────┘
selectors通常用于更复杂的 Select --如所有字符串列、以特定短语开头的列等等.
正如在select()文档中所写的，您可以在那里找到列的路径列表或表达式.
所以这些都行得通--.select('nrs', 'random')、.select(pl.col('nrs', 'random'))、.select(pl.col('nrs'), pl.col('random')).