python - 如何创建具有多个替代值的掩码 (Pandas DataFrame)

Question

假设我们有一个 Pandas DataFrame f 定义如下。我正在尝试创建一个掩码以选择列“xx”中值为“a”或“b”的所有行(我想选择第 0、1、3、4 行)。

f = pd.DataFrame([['a', 'b','c','a', 'b','c'],['1', '2','3', '4', '5','6', ]])
f = f.transpose()
f.columns = ['xx', 'yy']
f

   xx   yy
0   a   1
1   b   2
2   c   3
3   a   4
4   b   5
5   c   6

在 pandas 中有什么优雅的方法可以做到这一点吗？我知道用 f.xx =='a' 选择所有行，我们可以做 f[f.xx == 'a']。虽然我还没有弄清楚如何选择 f.xx 是 'a' 或 'b' 的行。谢谢。

Answer 1

你可以使用 isin

 print(f[(f["xx"].isin(("a","b")))])

这会给你:

  xx yy
0  a  1
1  b  2
3  a  4
4  b  5

如果你真的想要一个面具，你可以使用或|:

mask = (f["xx"] == "a") | (f["xx"] == "b")

print(f[mask])

这会给你相同的输出:

 xx yy
0  a  1
1  b  2
3  a  4
4  b  5