python - Pandas 分类平等测试

Question

我使用的是 pandas 0.24.2，无法对 Categorical 执行简单的等式测试实例。例如，使用此代码:

d = DataFrame({"x" : [1, 2, 3, 4]})
d["y"] = pd.cut(d["x"], bins=3)

d 的内容现在是:

    x   y
0   1   (0.997, 2.0]
1   2   (0.997, 2.0]
2   3   (2.0, 3.0]
3   4   (3.0, 4.0]

但是，我无法测试 y 是否与字符串值相等，例如:

d["y"] == '(3.0, 4.0]'

产生:

0    False
1    False
2    False
3    False
Name: y, dtype: bool

由于 cut() 的输出，我知道 y 的基础类型是 Categorical .但是，我在 Categorical 中找不到任何特殊的相等方法；此外，该文档甚至指出“平等比较适用于……标量”。我的方法中缺少什么？

Answer 1

这是间隔，所以传递给 pd.Interval

d.y==pd.Interval(3,4)
Out[255]: 
0    False
1    False
2    False
3     True
Name: y, dtype: bool