对 pandas dataframe 的多列进行条件过滤

Question

我有一个 pandas dataframe 有三列，A（日期）、B（分类值）和 C（实际值）。

A     |  B   |  C
01-19 |  5   |  0.34
01-19 |  3   |  0.25
01-19 |  7   |  0.07
01-20 |  5   |  0.15
01-20 |  2   |  0.36

等等。

我想要的是根据日期和特定阈值过滤行 - 例如：

[(01-19, x<6), (01-20, x<3)]

在这种情况下，那会给我

A     |  B   |  C
01-19 |  5   |  0.34
01-19 |  3   |  0.25
01-20 |  2   |  0.36

---

我的解决方案是用 A 和 B 建立一个多索引，但是我不完全确定如何过滤 B。

Answer 1

想法是为A列的所有值创建阈值字典，然后Series.map到新系列，因此可以通过B列进行比较并通过boolean indexing进行过滤：

d = {'01-19': 6, '01-20' : 3}
df = df[df['B'] < df['A'].map(d)]
print (df)
       A  B     C
0  01-19  5  0.34
1  01-19  3  0.25
4  01-20  2  0.36

详情：

print (df['A'].map(d))
0    6
1    6
2    6
3    3
4    3
Name: A, dtype: int64

Answer 2

如果只有两个条件，您可以像这样“和”和“或”运算符：

df = df[((df['A'] == '01-19') & (df['B'] < 6)) | ((df['A'] == '01-20') & (df['B'] < 3))]