Pandas 0.24.2中Categorical类型与字符串相等性测试失败问题求助
解决pandas 0.24.2中Categorical类型与字符串相等性测试的问题
你遇到的问题核心在于:pd.cut()返回的Categorical列中的元素并不是字符串,而是Interval对象,所以直接和字符串字面量比较自然会全部返回False。下面给你几个可行的解决方案:
方法1:将Categorical转换为字符串后比较
把列的类型强制转为字符串,再和目标字符串对比,这是最直观的方式:
d["y"].astype(str) == '(3.0, 4.0]'
执行后会得到正确的结果:
0 False 1 False 2 False 3 True Name: y, dtype: bool
方法2:直接与Interval对象比较
既然Categorical的元素是Interval,那就直接构造对应的Interval实例来做比较(注意pd.cut()默认是右闭合区间,所以构造时要指定closed='right'):
d["y"] == pd.Interval(3.0, 4.0, closed='right')
这个方法不需要类型转换,效率会更高一些。
方法3:使用isin()方法匹配标签
Categorical类型支持用isin()直接传入字符串形式的区间标签来筛选,这个方法也能达到目的:
d["y"].isin(['(3.0, 4.0]'])
返回的结果同样是正确的布尔序列。
补充说明
文档里提到的“相等性比较可与标量一起使用”,这里的标量指的是同类型的Interval对象,而不是字符串标量,这也是你之前误解的地方。
内容的提问来源于stack exchange,提问作者sparc_spread
相关产品推荐
相关产品推荐

