Polars使用布尔掩码按行过滤DataFrame却触发列过滤错误的问题咨询
Polars使用布尔掩码按行过滤DataFrame却触发列过滤错误的问题咨询
我最近在处理Polars DataFrame的时候遇到了个头疼的问题:我想用布尔掩码来按行过滤数据,就像在Pandas里常用的那种写法:
df = df[df['col'] == cond]
结果运行后直接报错了,错误提示说这个操作是用来选择列的:
expected xx values when selecting columns by boolean mask, got yy
这里的xx是我DataFrame的总列数,yy则是我生成的布尔掩码里True的数量。
我查了Polars的API文档,上面说这个方括号索引的语法应该是支持行过滤的,和Pandas的逻辑一致,但现在它却默认把这个布尔掩码当成列选择的条件了。
有没有什么办法能修改这个默认行为,让方括号索引按照我预期的那样处理行过滤呢?
PS:提前说明下,我不想用.filter或者.sql方法来实现,所以麻烦不要给我这方面的建议啦。
备注:内容来源于stack exchange,提问作者Ghost
相关产品推荐
相关产品推荐

