在 pandas 中过滤数据框

Question

我有一个数据框如下：

dic ={'wteam': [2, 3, 4, 2, 4], 'lteam': [3, 4, 2, 4, 2]}
pd.DataFrame(dic) 

   lteam  wteam
0    3      2
1    4      3
2    2      4
3    4      2
4    3      4

我需要一个新的数据框，它在 lteam 或 wteam 中有 2 个。

        lteam  wteam
    0    3      2
    2    2      4
    3    4      2

如何在 pandas 中执行此操作？

Answer 1

你的起始 df 的输出是错误的，最后一行应该是 [2,4]，除此之外我们可以调用 loc 对布尔过滤 df 生成的索引加上 drop any NaN 值：

In [15]:

df.loc[df[df==2].dropna(thresh=1).index]
Out[15]:
   lteam  wteam
0      3      2
2      2      4
3      4      2
4      2      4

分解：

In [16]:

df[df==2]
Out[16]:
   lteam  wteam
0    NaN      2
1    NaN    NaN
2      2    NaN
3    NaN      2
4      2    NaN
In [17]:

df[df==2].dropna(thresh=1)
Out[17]:
   lteam  wteam
0    NaN      2
2      2    NaN
3    NaN      2
4      2    NaN

更简洁的方法是提供 2 个布尔条件：

In [18]:

df[(df.lteam == 2) | (df.wteam == 2)]
Out[18]:
   lteam  wteam
0      3      2
2      2      4
3      4      2
4      2      4

由于运算符优先级，这需要使用按位 | 运算符和括号括起条件

如果你有很多列，第一种方法会更好，但对于你的简单数据集，后一种方法就可以了。

在 pandas 中过滤数据框

filtering a data frame in pandas

python

python-2.7

pandas