如何在 Polars 中按数据类型 select 列？

Question

在 pandas 中，我们有 pandas.DataFrame.select_dtypes 方法，它根据 dtype 选择某些列。在 Polars 中有没有类似的方法来做这样的事情？

Answer 1

可以将数据类型传递给 pl.col:

import polars as pl

df = pl.DataFrame(
    {
        "id": [1, 2, 3],
        "name": ["John", "Jane", "Jake"],
        "else": [10.0, 20.0, 30.0],
    }
)
print(df.select([pl.col(pl.Utf8), pl.col(pl.Int64)]))

输出：

shape: (3, 2)
┌──────┬─────┐
│ name ┆ id  │
│ ---  ┆ --- │
│ str  ┆ i64 │
╞══════╪═════╡
│ John ┆ 1   │
├╌╌╌╌╌╌┼╌╌╌╌╌┤
│ Jane ┆ 2   │
├╌╌╌╌╌╌┼╌╌╌╌╌┤
│ Jake ┆ 3   │
└──────┴─────┘

如何在 Polars 中按数据类型 select 列？

How to select columns by data type in Polars?

dataframe

python-polars