根据两个条件选择python数据框中的行

Question

基于下面的数据框（1），我希望创建一个数据框（2），其中y或z等于2.是否有方法可以方便地执行此操作？

如果我要创建一个仅包含dataframe（1）但不包含dataframe（2）的行的数据框（3），我该如何处理它？

id x y z 
    0 324 1 2 
    1 213 1 1 
    2 529 2 1 
    3 347 3 2 
    4 109 2 2 

...

Answer 1

您可以轻松创建df2足够使用条件：

df2 = df1[df1.y.eq(2) | df1.z.eq(2)] 

df2 
     x y z 
id   
0 324 1 2 
2 529 2 1 
3 347 3 2 
4 109 2 2 

鉴于df2和df1，你可以对指数进行组差异操作，如此：

df3 = df1.iloc[df1.index.difference(df2.index)] 

df3 
     x y z 
id   
1 213 1 1 

Answer 2

你可以做到以下几点：

import pandas as pd 

df = pd.read_csv('data.csv') 
df2 = df[(df.y == 2) | (df.z == 2)] 

print(df2) 

结果：

id x y z 
0 0 324 1 2 
2 2 529 2 1 
3 3 347 3 2 
4 4 109 2 2 

Answer 3

df[df[['y','z']].eq(2).any(1)] 
Out[1205]: 
    id x y z 
0 0 324 1 2 
2 2 529 2 1 
3 3 347 3 2 
4 4 109 2 2