python - Dataframe 过滤列(如果它是区间数据类型)

Question

我有一个数据框数组，其中包含一些列，其中之一是“时间”我想过滤时间在特定时间间隔内的行

为了简化问题，我制作了一个具有整数值和整数间隔的数据框

data=pd.DataFrame({'A':[1,2,3,4,5,6,7,8,9]})
interval=pd.Interval(1,4)
data[data['A'] in interval]

显示:ValueError:Series 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。

但预期结果是[2,3,4]

Answer 1

之间<​​

pandas.Series.between将返回可用于过滤数据帧的 bool 掩码。

data[data.A.between(1, 4)]

   A
0  1
1  2
2  3
3  4

---

查询

pandas.DataFrame.query也可以用来过滤

data.query('1 <= A <= 4')

   A
0  1
1  2
2  3
3  4

---

是在

pandas.Series.isin可以在使用 range 的特殊整数情况下使用。与 between 类似，我们创建了一个 bool 掩码。

data[data.A.isin(range(1, 5))]

   A
0  1
1  2
2  3
3  4