我有一个Polars DataFrame:
import polars as pl
import numpy as np
df = pl.DataFrame({'A': ['red', 'blue', 'green', np.nan, 'orange']})
my_list = ['red','orange']
我想知道my_list有哪些 colored颜色 .
对于Pandas ，我会做一些类似的事情:
df.A.isin(my_list)
但是，我得到了这个错误:
AttributeError: 'DataFrame' object has no attribute 'A'
如何在Polars中做到这一点？
以下代码可用于创建数据集，其中列A由指示每一项是否在my_list中的布尔列替换.
df.with_columns(pl.col("A").is_in(my_list))
shape: (5, 1)
┌───────┐
│ A │
│ --- │
│ bool │
╞═══════╡
│ true │
│ false │
│ false │
│ null │
│ true │
└───────┘
如果相反，你想
create一个新的布尔列，使用pl.Expr.alias如下
df.with_columns(pl.col("A").is_in(my_list).alias("B")
让返回值为pl.Series(就像@KrisKruse的答案中一样)，使用pl.DataFrame.get_column，如下所示
df.get_column("A").is_in(my_list)
filter对于值A为my_list的行，使用pl.DataFrame.filter，如下所示
df.filter(pl.col("A").is_in(my_list))